Merge pull request #805 from microsoft/bugfix-catmem-add-drop-function

[catmem] Add Drop Function
microsoft · Jun 30, 2023 · ef60c3d · ef60c3d
2 parents 3def8be + f8aad6c
commit ef60c3d
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");
+            }
+        }
+    }
+}