rt: Add constructors and destructors for stacks

2 files changed, 21 insertions, 3 deletions

diff --git a/src/rt/rust_stack.h b/src/rt/rust_stack.h
index d665b4fa2c1..34d4dda742d 100644
--- a/src/rt/rust_stack.h
+++ b/src/rt/rust_stack.h
@@ -1,3 +1,6 @@
+#ifndef RUST_STACK_H
+#define RUST_STACK_H
+
 struct stk_seg {
     stk_seg *prev;
     stk_seg *next;
@@ -10,6 +13,19 @@ struct stk_seg {
     uint8_t data[];
 };
 
+template <class T>
+stk_seg *
+create_stack(T allocer, size_t sz) {
+  size_t total_sz = sizeof(stk_seg) + sz;
+  return (stk_seg *)allocer->malloc(total_sz, "stack");
+}
+
+template <class T>
+void
+destroy_stack(T allocer, stk_seg *stk) {
+  allocer->free(stk);
+}
+
 void
 config_valgrind_stack(stk_seg *stk);
 
@@ -21,3 +37,5 @@ add_stack_canary(stk_seg *stk);
 
 void
 check_stack_canary(stk_seg *stk);
+
+#endif /* RUST_STACK_H */
diff --git a/src/rt/rust_task.cpp b/src/rt/rust_task.cpp
index dc8b911778a..32245c5d9cd 100644
--- a/src/rt/rust_task.cpp
+++ b/src/rt/rust_task.cpp
@@ -538,7 +538,7 @@ void
 rust_task::free_stack(stk_seg *stk) {
     LOGPTR(thread, "freeing stk segment", (uintptr_t)stk);
     total_stack_sz -= user_stack_size(stk);
-    free(stk);
+    destroy_stack(this, stk);
 }
 
 void
@@ -581,8 +581,8 @@ rust_task::new_stack(size_t requested_sz) {
         fail();
     }
 
-    size_t sz = sizeof(stk_seg) + rust_stk_sz + RED_ZONE_SIZE;
-    stk_seg *new_stk = (stk_seg *)malloc(sz, "stack");
+    size_t sz = rust_stk_sz + RED_ZONE_SIZE;
+    stk_seg *new_stk = create_stack(this, sz);
     LOGPTR(thread, "new stk", (uintptr_t)new_stk);
     memset(new_stk, 0, sizeof(stk_seg));
     add_stack_canary(new_stk);