Introducing the fact we need a more efficient stack allocator, and that it's to be performed from the task manager.

2 files changed, 18 insertions, 2 deletions

diff --git a/src/Task.cc b/src/Task.cc
index b512e23..d72decc 100644
--- a/src/Task.cc
+++ b/src/Task.cc
@@ -13,10 +13,18 @@ Balau::Task::Task() {
     Printer::elog(E_TASK, "Created a Task at %p");
 }
 
+bool Balau::Task::needsStacks() {
+#ifndef _WIN32
+    return true;
+#else
+    return false;
+#endif
+}
+
 void Balau::Task::setup(TaskMan * taskMan) {
     size_t size = stackSize();
 #ifndef _WIN32
-    m_stack = malloc(size);
+    m_stack = taskMan->getStack();
     coro_create(&m_ctx, coroutineTrampoline, this, m_stack, size);
 #else
     m_stack = NULL;
@@ -34,7 +42,7 @@ void Balau::Task::setup(TaskMan * taskMan) {
 
 Balau::Task::~Task() {
     if (m_stack)
-        free(m_stack);
+        m_taskMan->freeStack(m_stack);
     free(m_tls);
 }
 
diff --git a/src/TaskMan.cc b/src/TaskMan.cc
index 375a94d..e9e3485 100644
--- a/src/TaskMan.cc
+++ b/src/TaskMan.cc
@@ -134,6 +134,14 @@ Balau::TaskMan::~TaskMan() {
     ev_loop_destroy(m_loop);
 }
 
+void * Balau::TaskMan::getStack() {
+    return malloc(Task::stackSize());
+}
+
+void Balau::TaskMan::freeStack(void * stack) {
+    free(stack);
+}
+
 void Balau::TaskMan::mainLoop() {
     // We need at least one round before bailing :)
     do {