From 09708cd98486578168d01c043a053941ab378b2c Mon Sep 17 00:00:00 2001
From: Alwin Joshy <joshyalwin@gmail.com>
Date: Fri, 10 May 2024 13:19:54 +1000
Subject: [PATCH] Add TCB_SetSpace invocation

Signed-off-by: Alwin Joshy <joshyalwin@gmail.com>
---
 crates/sel4/src/invocations.rs | 27 ++++++++++++++++++++++-----
 1 file changed, 22 insertions(+), 5 deletions(-)

diff --git a/crates/sel4/src/invocations.rs b/crates/sel4/src/invocations.rs
index 14c0bf07b..b2412eae2 100644
--- a/crates/sel4/src/invocations.rs
+++ b/crates/sel4/src/invocations.rs
@@ -11,16 +11,13 @@ use core::mem;
 use sel4_config::{sel4_cfg, sel4_cfg_if};
 
 use crate::{
-    cap::*, sys, AbsoluteCPtr, CNodeCapData, CapRights, Error, InvocationContext, ObjectBlueprint,
-    Result, UserContext, Word,
+    cap::*, ipc_buffer, sys, AbsoluteCPtr, CNodeCapData, CPtr, CapRights, Error, InvocationContext,
+    ObjectBlueprint, Result, UserContext, Word,
 };
 
 #[sel4_cfg(KERNEL_MCS)]
 use crate::Badge;
 
-#[sel4_cfg(not(KERNEL_MCS))]
-use crate::CPtr;
-
 /// Corresponds to `seL4_Time`.
 #[sel4_cfg(KERNEL_MCS)]
 pub type Time = u64;
@@ -159,6 +156,26 @@ impl<C: InvocationContext> Tcb<C> {
         }
     }
 
+    /// Corresponds to `seL4_TCB_SetSpace`.
+    pub fn tcb_set_space(
+        self,
+        fault_ep: CPtr,
+        cspace_root: CNode,
+        cspace_root_data: CNodeCapData,
+        vspace_root: VSpace,
+    ) -> Result<()> {
+        Error::wrap(self.invoke(|cptr, ipc_buffer| {
+            ipc_buffer.inner_mut().seL4_TCB_SetSpace(
+                cptr.bits(),
+                fault_ep.bits(),
+                cspace_root.bits(),
+                cspace_root_data.into_word(),
+                vspace_root.bits(),
+                0, /* HACK */
+            )
+        }))
+    }
+
     sel4_cfg_if! {
         if #[sel4_cfg(KERNEL_MCS)] {
             /// Corresponds to `seL4_TCB_SetSchedParams`.