zephyr: memdomain api

Author: tylerwhall

rust-app/zephyr/src/lib.rs

@@ -6,6 +6,7 @@ extern crate derive_more;
 
 pub mod device;
 pub mod kobj;
+pub mod memdomain;
 pub mod mempool;
 pub mod mutex;
 pub mod semaphore;

rust-app/zephyr/src/memdomain.rs

@@ -0,0 +1,61 @@
+#[cfg(not(usermode))]
+use core::marker::PhantomData;
+
+pub use zephyr_sys::raw::k_mem_domain;
+
+use crate::thread::ThreadId;
+
+#[cfg(usermode)]
+pub struct MemDomain<'a>(&'a k_mem_domain);
+#[cfg(not(usermode))]
+pub struct MemDomain<'a>(PhantomData<&'a ()>);
+
+impl<'a> MemDomain<'a> {
+    #[cfg(usermode)]
+    pub unsafe fn new(domain: &'a k_mem_domain) -> Self {
+        MemDomain(domain)
+    }
+    #[cfg(not(usermode))]
+    pub fn new() -> Self {
+        MemDomain(PhantomData)
+    }
+
+    pub fn add_thread<C: MemDomainAPI>(&self, _thread: ThreadId) {
+        #[cfg(usermode)]
+        C::k_mem_domain_add_thread(self.0, _thread)
+    }
+}
+
+pub trait MemDomainAPI {
+    fn k_mem_domain_add_thread(domain: &k_mem_domain, thread: ThreadId);
+}
+
+impl MemDomainAPI for crate::context::Kernel {
+    fn k_mem_domain_add_thread(domain: &k_mem_domain, thread: ThreadId) {
+        unsafe {
+            zephyr_sys::raw::k_mem_domain_add_thread(domain as *const _ as *mut _, thread.tid())
+        }
+    }
+}
+
+/// Get a static reference to an external mem domain
+#[cfg(usermode)]
+#[macro_export]
+macro_rules! static_mem_domain {
+    ($domain:ident) => {{
+        extern "C" {
+            #[no_mangle]
+            static $domain: $crate::memdomain::k_mem_domain;
+        }
+
+        unsafe { $crate::memdomain::MemDomain::new(&$domain) }
+    }};
+}
+/// Get a static reference to an external mem domain
+#[cfg(not(usermode))]
+#[macro_export]
+macro_rules! static_mem_domain {
+    ($domain:ident) => {{
+        $crate::memdomain::MemDomain::new()
+    }};
+}

rust-app/zephyr/src/thread.rs

@@ -4,6 +4,10 @@ use crate::kobj::KObj;
 pub struct ThreadId(zephyr_sys::raw::k_tid_t);
 
 impl ThreadId {
+    pub fn tid(&self) -> zephyr_sys::raw::k_tid_t {
+        self.0
+    }
+
     pub fn k_object_access_grant<C: ThreadSyscalls, K: KObj>(&self, kobj: &K) {
         C::k_object_access_grant(kobj, *self)
     }