lib: monkey: Test to add allocator trace

Signed-off-by: Hiroshi Hatake <hiroshi@chronosphere.io>

Author: cosmo0920

lib/monkey/deps/flb_libco/amd64.c

@@ -117,6 +117,56 @@ static void (*co_swap)(cothread_t, cothread_t) = 0;
   }
 #endif
 
+
+struct co_allocation {
+  cothread_t handle;
+  struct co_allocation *next;
+};
+
+static thread_local struct co_allocation *co_allocations;
+
+static int co_allocation_track(cothread_t handle) {
+  struct co_allocation *entry;
+
+  entry = (struct co_allocation *)malloc(sizeof(struct co_allocation));
+  if (entry == 0) {
+    return 0;
+  }
+
+  entry->handle = handle;
+  entry->next = co_allocations;
+  co_allocations = entry;
+
+  return 1;
+}
+
+static int co_allocation_untrack(cothread_t handle) {
+  struct co_allocation *entry;
+  struct co_allocation *prev;
+
+  prev = 0;
+  entry = co_allocations;
+
+  while (entry != 0) {
+    if (entry->handle == handle) {
+      if (prev != 0) {
+        prev->next = entry->next;
+      }
+      else {
+        co_allocations = entry->next;
+      }
+
+      free(entry);
+      return 1;
+    }
+
+    prev = entry;
+    entry = entry->next;
+  }
+
+  return 0;
+}
+
 static void crash() {
   assert(0);  /* called only if cothread_t entrypoint returns */
 }
@@ -144,12 +194,25 @@ cothread_t co_create(unsigned int size, void (*entrypoint)(void),
     *--p = (long long)crash;                            /* crash if entrypoint returns */
     *--p = (long long)entrypoint;                       /* start of function */
     *(long long*)handle = (long long)p;                 /* stack pointer */
+
+    if (!co_allocation_track(handle)) {
+      free(handle);
+      return 0;
+    }
   }
 
   return handle;
 }
 
 void co_delete(cothread_t handle) {
+  if (handle == 0) {
+    return;
+  }
+
+  if (!co_allocation_untrack(handle)) {
+    return;
+  }
+
   free(handle);
 }