[catmem] Bug Fix: Add drop function

microsoft · Jun 30, 2023 · f8aad6c · f8aad6c
1 parent 3def8be
commit f8aad6c
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/rust/catmem/mod.rs b/src/rust/catmem/mod.rs
@@ -519,3 +519,19 @@ impl CatmemLibOS {
         self.scheduler.poll()
     }
 }
+
+//======================================================================================================================
+// Trait Implementations
+//======================================================================================================================
+
+impl Drop for CatmemLibOS {
+    // Releases all sockets allocated by Catnap.
+    fn drop(&mut self) {
+        for (_, queue) in self.qtable.borrow().get_values() {
+            if let Err(e) = push_eof(queue.get_pipe().buffer()) {
+                error!("push_eof() failed: {:?}", e);
+                warn!("leaking shared memory region");
+            }
+        }
+    }
+}