Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
try to group the mallocs by finality instead of spreading data all around the memory
[simgrid.git] / src / xbt / mallocator.c
index b678466..1b0a3ce 100644 (file)
@@ -1,6 +1,7 @@
 /* mallocator - recycle objects to avoid malloc() / free()                  */
 
-/* Copyright (c) 2006 Christophe Thiery. All rights reserved.               */
+/* Copyright (c) 2006, 2007, 2008, 2009, 2010. The SimGrid Team.
+ * All rights reserved.                                                     */
 
 /* This program is free software; you can redistribute it and/or modify it
  * under the terms of the license (GNU LGPL) which comes with this package. */
@@ -8,8 +9,11 @@
 #include "xbt/mallocator.h"
 #include "xbt/asserts.h"
 #include "xbt/sysdep.h"
+#include "mc/mc.h" /* kill mallocators when model-checking is enabled */
 #include "mallocator_private.h"
 
+XBT_LOG_NEW_DEFAULT_SUBCATEGORY(xbt_mallocator, xbt, "Mallocators");
+
 /**
  * \brief Constructor
  * \param size size of the internal stack: number of objects the mallocator
@@ -20,7 +24,7 @@
  * in \a xbt_mallocator_release() when the stack is full, and when
  * the mallocator is freed.
  * \param reset_f function to reinitialise an object of your datatype, called
- * when you extract an object from the mallocator 
+ * when you extract an object from the mallocator
  *
  * Create and initialize a new mallocator for a given datatype.
  *
  * \see xbt_mallocator_free()
  */
 xbt_mallocator_t xbt_mallocator_new(int size,
-                                   pvoid_f_void_t new_f,
-                                   void_f_pvoid_t free_f,
-                                   void_f_pvoid_t reset_f) {
+                                    pvoid_f_void_t new_f,
+                                    void_f_pvoid_t free_f,
+                                    void_f_pvoid_t reset_f)
+{
 
 
   xbt_mallocator_t m;
 
-  xbt_assert0(size > 0, "size must be positive");
-  xbt_assert0(new_f != NULL && free_f != NULL && reset_f != NULL,"invalid parameter");
+  xbt_assert(size > 0, "size must be positive");
+  xbt_assert(new_f != NULL && free_f != NULL
+              && reset_f != NULL, "invalid parameter");
+
+  /* Let's force 0 size mallocator! (Dirty hack, blame Martin :) ) */
+
+  /* mallocators and memory mess introduced by model-checking do not mix well together:
+   *   The mallocator will give standard memory when we are using raw memory (so these blocks are killed on restore)
+   *   and the contrary (so these blocks will leak accross restores)
+   */
+  if (MC_IS_ENABLED)
+    size = 0;
 
   m = xbt_new0(s_xbt_mallocator_t, 1);
+  XBT_VERB("Create mallocator %p", m);
+  if (XBT_LOG_ISENABLED(xbt_mallocator, xbt_log_priority_verbose))
+    xbt_backtrace_display_current();
 
-  m->objects = xbt_new0(void*, size);
+  m->objects = xbt_new0(void *, MC_IS_ENABLED ? 1 : size);
   m->max_size = size;
   m->current_size = 0;
   m->new_f = new_f;
@@ -58,14 +76,16 @@ xbt_mallocator_t xbt_mallocator_new(int size,
  *
  * \see xbt_mallocator_new()
  */
-void xbt_mallocator_free(xbt_mallocator_t m) {
+void xbt_mallocator_free(xbt_mallocator_t m)
+{
 
   int i;
-  xbt_assert0(m != NULL, "Invalid parameter");
-
+  xbt_assert(m != NULL, "Invalid parameter");
 
+  XBT_VERB("Frees mallocator %p (size:%d/%d)", m, m->current_size,
+        m->max_size);
   for (i = 0; i < m->current_size; i++) {
-    m->free_f(m->objects[i]);
+    (*(m->free_f)) (m->objects[i]);
   }
   xbt_free(m->objects);
   xbt_free(m);
@@ -79,28 +99,40 @@ void xbt_mallocator_free(xbt_mallocator_t m) {
  * This function is designed to be used instead of malloc().
  * If the mallocator is not empty, an object is
  * extracted from the mallocator and no malloc is done.
- * 
+ *
  * If the mallocator is empty, a new object is created,
  * by calling the function new_f().
- * 
+ *
  * In both cases, the function reset_f() is called on the object.
  *
  * \see xbt_mallocator_release()
  */
-void *xbt_mallocator_get(xbt_mallocator_t m) {
+void *xbt_mallocator_get(xbt_mallocator_t m)
+{
   void *object;
-  xbt_assert0(m != NULL, "Invalid parameter");
 
+  if (m->current_size <= 0) {
+    /* No object is ready yet. Create a bunch of them to try to group the mallocs
+     *  on the same memory pages (to help the cache lines) */
+
+    /* XBT_DEBUG("Create a new object for mallocator %p (size:%d/%d)", m,
+           m->current_size, m->max_size); */
+    int i;
+    int amount=MIN( (m->max_size) /2,1000);
+    for (i=0;i<amount;i++)
+      m->objects[i] = (*(m->new_f)) ();
+    m->current_size=amount;
+  }
 
-  if (m->current_size > 0) {
-    /* there is at least an available object */
+  /* there is at least an available object, now */
+  /* XBT_DEBUG("Reuse an old object for mallocator %p (size:%d/%d)", m,
+           m->current_size, m->max_size); */
+  if (MC_IS_ENABLED) /* no mallocator with MC */
+    object = (*(m->new_f)) ();
+  else
     object = m->objects[--m->current_size];
-  }
-  else {
-    /* otherwise we must allocate a new object */
-    object = m->new_f();
-  }
-  m->reset_f(object);
+
+  (*(m->reset_f)) (object);
   return object;
 }
 
@@ -117,15 +149,18 @@ void *xbt_mallocator_get(xbt_mallocator_t m) {
  *
  * \see xbt_mallocator_get()
  */
-void xbt_mallocator_release(xbt_mallocator_t m, void *object) {
-  xbt_assert0(m != NULL && object != NULL, "Invalid parameter");
-
+void xbt_mallocator_release(xbt_mallocator_t m, void *object)
+{
   if (m->current_size < m->max_size) {
     /* there is enough place to push the object */
+    /* XBT_DEBUG
+        ("Store deleted object in mallocator %p for further use (size:%d/%d)",
+         m, m->current_size, m->max_size); */
     m->objects[m->current_size++] = object;
-  }
-  else {
+  } else {
     /* otherwise we don't have a choice, we must free the object */
-    m->free_f(object);
+    /* XBT_DEBUG("Free deleted object: mallocator %p is full (size:%d/%d)", m,
+           m->current_size, m->max_size); */
+    (*(m->free_f)) (object);
   }
 }