syscall/obj: Close an ObjHandle when dropped

Implement Drop for the ObjHandle to close the underlying kernel object when the ObjHandle is dropped in the user mode. Co-developed-by: Vijay Dhanraj <[email protected]> Signed-off-by: Chuanxiao Dong <[email protected]>
coconut-svsm · Nov 6, 2024 · 8c00c05 · 8c00c05
1 parent 1fc0b60
commit 8c00c05
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/syscall/src/obj.rs b/syscall/src/obj.rs
@@ -4,6 +4,9 @@
 //
 // Author: Chuanxiao Dong <[email protected]>
 
+use super::call::syscall1;
+use super::SYS_CLOSE;
+
 /// The object is exposed to the user mode via the object-opening related
 /// syscalls, which returns the id of the object created by the COCONUT-SVSM
 /// kernel. The user mode can make use this id to access the corresponding
@@ -31,3 +34,11 @@ impl From<&ObjHandle> for u32 {
 pub trait Obj {
     fn id(&self) -> u32;
 }
+
+impl Drop for ObjHandle {
+    fn drop(&mut self) {
+        unsafe {
+            let _ = syscall1(SYS_CLOSE, self.0.into());
+        }
+    }
+}