save
[protos/libecoli.git] / lib / ecoli_malloc.c
index 0bdbf21..3113f0e 100644 (file)
  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
-
+#include <ecoli_init.h>
 #include <ecoli_malloc.h>
 
 #include <stdlib.h>
 #include <string.h>
 #include <errno.h>
 
+static int init_done = 0;
+
 struct ec_malloc_handler ec_malloc_handler;
 
 int ec_malloc_register(ec_malloc_t usr_malloc, ec_free_t usr_free,
        ec_realloc_t usr_realloc)
 {
-       if (usr_malloc == NULL || usr_free == NULL || usr_realloc == NULL)
+       if (usr_malloc == NULL || usr_free == NULL || usr_realloc == NULL) {
+               errno = EINVAL;
+               return -1;
+       }
+
+       if (init_done) {
+               errno = EBUSY;
                return -1;
+       }
 
        ec_malloc_handler.malloc = usr_malloc;
        ec_malloc_handler.free = usr_free;
@@ -47,16 +56,14 @@ int ec_malloc_register(ec_malloc_t usr_malloc, ec_free_t usr_free,
        return 0;
 }
 
-void ec_malloc_unregister(void)
+void *__ec_malloc(size_t size, const char *file, unsigned int line)
 {
-       ec_malloc_handler.malloc = NULL;
-       ec_malloc_handler.free = NULL;
-       ec_malloc_handler.realloc = NULL;
+       return ec_malloc_handler.malloc(size, file, line);
 }
 
-void *__ec_malloc(size_t size, const char *file, unsigned int line)
+void *ec_malloc_func(size_t size)
 {
-       return ec_malloc_handler.malloc(size, file, line);
+       return __ec_malloc(size, __FILE__, __LINE__);
 }
 
 void __ec_free(void *ptr, const char *file, unsigned int line)
@@ -64,12 +71,18 @@ void __ec_free(void *ptr, const char *file, unsigned int line)
        ec_malloc_handler.free(ptr, file, line);
 }
 
+void ec_free_func(void *ptr)
+{
+       __ec_free(ptr, __FILE__, __LINE__);
+}
+
 void *__ec_calloc(size_t nmemb, size_t size, const char *file,
        unsigned int line)
 {
        void *ptr;
        size_t total;
 
+       /* check overflow */
        total = size * nmemb;
        if (nmemb != 0 && size != (total / nmemb)) {
                errno = ENOMEM;
@@ -80,7 +93,7 @@ void *__ec_calloc(size_t nmemb, size_t size, const char *file,
        if (ptr == NULL)
                return NULL;
 
-       memset(ptr, 0, size);
+       memset(ptr, 0, total);
        return ptr;
 }
 
@@ -117,3 +130,16 @@ char *__ec_strndup(const char *s, size_t n, const char *file, unsigned int line)
 
        return s2;
 }
+
+static int ec_malloc_init_func(void)
+{
+       init_done = 1;
+       return 0;
+}
+
+static struct ec_init ec_malloc_init = {
+       .init = ec_malloc_init_func,
+       .priority = 50,
+};
+
+EC_INIT_REGISTER(ec_malloc_init);