r4473: - moved talloc into its own lib/talloc/ area
authorAndrew Tridgell <tridge@samba.org>
Sun, 2 Jan 2005 07:47:34 +0000 (07:47 +0000)
committerGerald (Jerry) Carter <jerry@samba.org>
Wed, 10 Oct 2007 18:07:55 +0000 (13:07 -0500)
- added gcov flags to Makefile.talloc

- expanded talloc testsuite to add a test for realloc with a child ptr

- fixed a bug in talloc_realloc() with realloc of a ptr that has child ptrs
(This used to be commit 98b5f73c1ba34d7576c5995069b485c1c5ede324)

12 files changed:
source4/configure.in
source4/include/includes.h
source4/lib/basic.mk
source4/lib/ldb/common/talloc.c [deleted file]
source4/lib/talloc/Makefile.talloc [new file with mode: 0644]
source4/lib/talloc/config.m4 [new file with mode: 0644]
source4/lib/talloc/config.mk [new file with mode: 0644]
source4/lib/talloc/talloc.c [moved from source4/lib/talloc.c with 99% similarity]
source4/lib/talloc/talloc.h [moved from source4/include/talloc.h with 98% similarity]
source4/lib/talloc/talloc_guide.txt [moved from talloc_guide.txt with 100% similarity]
source4/lib/talloc/testsuite.c [moved from source4/torture/local/talloc.c with 91% similarity]
source4/torture/config.mk

index 241d154ae14cc2af8f0c3b36b418f57773be9732..7672ed5eb23c5fbc1d6ee4ac3db27ebe446caec0 100644 (file)
@@ -14,6 +14,7 @@ SMB_INCLUDE_M4(build/m4/rewrite.m4)
 SMB_INCLUDE_M4(lib/popt/config.m4)
 SMB_INCLUDE_M4(lib/iconv.m4)
 SMB_INCLUDE_M4(lib/socket/config.m4)
+SMB_INCLUDE_M4(lib/talloc/config.m4)
 SMB_INCLUDE_M4(lib/tdb/config.m4)
 SMB_INCLUDE_M4(lib/ldb/config.m4)
 SMB_INCLUDE_M4(lib/cmdline/config.m4)
index ad24ff676d5074b697a9aff67636922af6b35c78..2666d59f898cdf54c37c65630ea84741c797f464 100644 (file)
@@ -154,7 +154,7 @@ extern int errno;
 /* Lists, trees, caching, database... */
 #include "version.h"
 #include "xfile.h"
-#include "talloc.h"
+#include "lib/talloc/talloc.h"
 #include "nt_status.h"
 #include "structs.h"
 #include "lib/tdb/include/tdb.h"
@@ -291,8 +291,10 @@ int asprintf(char **,const char *, ...) PRINTF_ATTRIBUTE(2,3);
 #define strlen(x) valgrind_strlen(x)
 #endif
 
+#if 0
+/* darn, we can't do this now that we don't link the ldb tools to all the smb libs */
 #define TALLOC_ABORT(reason) smb_panic(reason)
-
+#endif
 
 /*
   this is a warning hack. The idea is to use this everywhere that we
index e4275bea49a55d9afeac3019b5c7d9e7612d153c..a14201d9be671e7bd5d20261d13a6f0619df6bec 100644 (file)
@@ -56,7 +56,6 @@ ADD_OBJ_FILES = \
                lib/data_blob.o \
                lib/util.o \
                lib/util_sock.o \
-               lib/talloc.o \
                lib/substitute.o \
                lib/fsusage.o \
                lib/ms_fnmatch.o \
@@ -71,6 +70,6 @@ ADD_OBJ_FILES = \
                lib/db_wrap.o \
                lib/gencache.o
 REQUIRED_SUBSYSTEMS = \
-               LIBLDB CHARSET LIBREPLACE LIBNETIF LIBCRYPTO LIB_SECURITY EXT_LIB_DL
+               LIBLDB CHARSET LIBREPLACE LIBNETIF LIBCRYPTO LIB_SECURITY EXT_LIB_DL LIBTALLOC
 # End SUBSYSTEM LIBBASIC
 ##############################
diff --git a/source4/lib/ldb/common/talloc.c b/source4/lib/ldb/common/talloc.c
deleted file mode 100644 (file)
index 7f50264..0000000
+++ /dev/null
@@ -1,997 +0,0 @@
-/* 
-   Samba Unix SMB/CIFS implementation.
-
-   Samba trivial allocation library - new interface
-
-   NOTE: Please read talloc_guide.txt for full documentation
-
-   Copyright (C) Andrew Tridgell 2004
-   
-   This program is free software; you can redistribute it and/or modify
-   it under the terms of the GNU General Public License as published by
-   the Free Software Foundation; either version 2 of the License, or
-   (at your option) any later version.
-   
-   This program is distributed in the hope that it will be useful,
-   but WITHOUT ANY WARRANTY; without even the implied warranty of
-   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-   GNU General Public License for more details.
-   
-   You should have received a copy of the GNU General Public License
-   along with this program; if not, write to the Free Software
-   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
-*/
-
-/*
-  inspired by http://swapped.cc/halloc/
-*/
-
-
-#ifdef _SAMBA_BUILD_
-#include "includes.h"
-#else
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
-#include <stdarg.h>
-#include "talloc.h"
-#endif
-
-/* use this to force every realloc to change the pointer, to stress test
-   code that might not cope */
-#define ALWAYS_REALLOC 0
-
-
-#define MAX_TALLOC_SIZE 0x10000000
-#define TALLOC_MAGIC 0xe814ec4f
-#define TALLOC_MAGIC_FREE 0x7faebef3
-#define TALLOC_MAGIC_REFERENCE ((const char *)1)
-
-/* by default we abort when given a bad pointer (such as when talloc_free() is called 
-   on a pointer that came from malloc() */
-#ifndef TALLOC_ABORT
-#define TALLOC_ABORT(reason) abort()
-#endif
-
-#ifndef discard_const_p
-#define discard_const_p(type, ptr) ((type *)(ptr))
-#endif
-
-/* this null_context is only used if talloc_enable_leak_report() or
-   talloc_enable_leak_report_full() is called, otherwise it remains
-   NULL
-*/
-static const void *null_context;
-
-
-struct talloc_reference_handle {
-       struct talloc_reference_handle *next, *prev;
-       void *ptr;
-};
-
-typedef int (*talloc_destructor_t)(void *);
-
-struct talloc_chunk {
-       struct talloc_chunk *next, *prev;
-       struct talloc_chunk *parent, *child;
-       struct talloc_reference_handle *refs;
-       size_t size;
-       unsigned magic;
-       talloc_destructor_t destructor;
-       const char *name;
-};
-
-/* panic if we get a bad magic value */
-static struct talloc_chunk *talloc_chunk_from_ptr(const void *ptr)
-{
-       struct talloc_chunk *tc = discard_const_p(struct talloc_chunk, ptr)-1;
-       if (tc->magic != TALLOC_MAGIC) { 
-               if (tc->magic == TALLOC_MAGIC_FREE) {
-                       TALLOC_ABORT("Bad talloc magic value - double free"); 
-               } else {
-                       TALLOC_ABORT("Bad talloc magic value - unknown value"); 
-               }
-       }
-
-       return tc;
-}
-
-/* hook into the front of the list */
-#define _TLIST_ADD(list, p) \
-do { \
-        if (!(list)) { \
-               (list) = (p); \
-               (p)->next = (p)->prev = NULL; \
-       } else { \
-               (list)->prev = (p); \
-               (p)->next = (list); \
-               (p)->prev = NULL; \
-               (list) = (p); \
-       }\
-} while (0)
-
-/* remove an element from a list - element doesn't have to be in list. */
-#define _TLIST_REMOVE(list, p) \
-do { \
-       if ((p) == (list)) { \
-               (list) = (p)->next; \
-               if (list) (list)->prev = NULL; \
-       } else { \
-               if ((p)->prev) (p)->prev->next = (p)->next; \
-               if ((p)->next) (p)->next->prev = (p)->prev; \
-       } \
-       if ((p) && ((p) != (list))) (p)->next = (p)->prev = NULL; \
-} while (0)
-
-
-/*
-  return the parent chunk of a pointer
-*/
-static struct talloc_chunk *talloc_parent_chunk(const void *ptr)
-{
-       struct talloc_chunk *tc = talloc_chunk_from_ptr(ptr);
-       while (tc->prev) tc=tc->prev;
-       return tc->parent;
-}
-
-/* 
-   Allocate a bit of memory as a child of an existing pointer
-*/
-void *_talloc(const void *context, size_t size)
-{
-       struct talloc_chunk *tc;
-
-       if (context == NULL) {
-               context = null_context;
-       }
-
-       if (size >= MAX_TALLOC_SIZE) {
-               return NULL;
-       }
-
-       tc = malloc(sizeof(*tc)+size);
-       if (tc == NULL) return NULL;
-
-       tc->size = size;
-       tc->magic = TALLOC_MAGIC;
-       tc->destructor = NULL;
-       tc->child = NULL;
-       tc->name = NULL;
-       tc->refs = NULL;
-
-       if (context) {
-               struct talloc_chunk *parent = talloc_chunk_from_ptr(context);
-
-               tc->parent = parent;
-
-               if (parent->child) {
-                       parent->child->parent = NULL;
-               }
-
-               _TLIST_ADD(parent->child, tc);
-       } else {
-               tc->next = tc->prev = tc->parent = NULL;
-       }
-
-       return (void *)(tc+1);
-}
-
-
-/*
-  setup a destructor to be called on free of a pointer
-  the destructor should return 0 on success, or -1 on failure.
-  if the destructor fails then the free is failed, and the memory can
-  be continued to be used
-*/
-void talloc_set_destructor(const void *ptr, int (*destructor)(void *))
-{
-       struct talloc_chunk *tc = talloc_chunk_from_ptr(ptr);
-       tc->destructor = destructor;
-}
-
-/*
-  increase the reference count on a piece of memory. 
-*/
-void talloc_increase_ref_count(const void *ptr)
-{
-       talloc_reference(null_context, ptr);
-}
-
-/*
-  helper for talloc_reference()
-*/
-static int talloc_reference_destructor(void *ptr)
-{
-       struct talloc_reference_handle *handle = ptr;
-       struct talloc_chunk *tc1 = talloc_chunk_from_ptr(ptr);
-       struct talloc_chunk *tc2 = talloc_chunk_from_ptr(handle->ptr);
-       if (tc1->destructor != (talloc_destructor_t)-1) {
-               tc1->destructor = NULL;
-       }
-       _TLIST_REMOVE(tc2->refs, handle);
-       talloc_free(handle);
-       return 0;
-}
-
-/*
-  make a secondary reference to a pointer, hanging off the given context.
-  the pointer remains valid until both the original caller and this given
-  context are freed.
-  
-  the major use for this is when two different structures need to reference the 
-  same underlying data, and you want to be able to free the two instances separately,
-  and in either order
-*/
-void *talloc_reference(const void *context, const void *ptr)
-{
-       struct talloc_chunk *tc;
-       struct talloc_reference_handle *handle;
-       if (ptr == NULL) return NULL;
-
-       tc = talloc_chunk_from_ptr(ptr);
-       handle = talloc_named_const(context, sizeof(*handle), TALLOC_MAGIC_REFERENCE);
-
-       if (handle == NULL) return NULL;
-
-       /* note that we hang the destructor off the handle, not the
-          main context as that allows the caller to still setup their
-          own destructor on the context if they want to */
-       talloc_set_destructor(handle, talloc_reference_destructor);
-       handle->ptr = discard_const_p(void, ptr);
-       _TLIST_ADD(tc->refs, handle);
-       return handle->ptr;
-}
-
-/*
-  remove a secondary reference to a pointer. This undo's what
-  talloc_reference() has done. The context and pointer arguments
-  must match those given to a talloc_reference()
-*/
-static int talloc_unreference(const void *context, const void *ptr)
-{
-       struct talloc_chunk *tc = talloc_chunk_from_ptr(ptr);
-       struct talloc_reference_handle *h;
-
-       if (context == NULL) {
-               context = null_context;
-       }
-
-       for (h=tc->refs;h;h=h->next) {
-               struct talloc_chunk *p = talloc_parent_chunk(h);
-               if ((p==NULL && context==NULL) || p+1 == context) break;
-       }
-       if (h == NULL) {
-               return -1;
-       }
-
-       talloc_set_destructor(h, NULL);
-       _TLIST_REMOVE(tc->refs, h);
-       talloc_free(h);
-       return 0;
-}
-
-/*
-  remove a specific parent context from a pointer. This is a more
-  controlled varient of talloc_free()
-*/
-int talloc_unlink(const void *context, void *ptr)
-{
-       struct talloc_chunk *tc_p, *new_p;
-       void *new_parent;
-
-       if (ptr == NULL) {
-               return -1;
-       }
-
-       if (context == NULL) {
-               context = null_context;
-       }
-
-       if (talloc_unreference(context, ptr) == 0) {
-               return 0;
-       }
-
-       if (context == NULL) {
-               if (talloc_parent_chunk(ptr) != NULL) {
-                       return -1;
-               }
-       } else {
-               if (talloc_chunk_from_ptr(context) != talloc_parent_chunk(ptr)) {
-                       return -1;
-               }
-       }
-       
-       tc_p = talloc_chunk_from_ptr(ptr);
-
-       if (tc_p->refs == NULL) {
-               return talloc_free(ptr);
-       }
-
-       new_p = talloc_parent_chunk(tc_p->refs);
-       if (new_p) {
-               new_parent = new_p+1;
-       } else {
-               new_parent = NULL;
-       }
-
-       if (talloc_unreference(new_parent, ptr) != 0) {
-               return -1;
-       }
-
-       talloc_steal(new_parent, ptr);
-
-       return 0;
-}
-
-/*
-  add a name to an existing pointer - va_list version
-*/
-static void talloc_set_name_v(const void *ptr, const char *fmt, va_list ap) PRINTF_ATTRIBUTE(2,0);
-
-static void talloc_set_name_v(const void *ptr, const char *fmt, va_list ap)
-{
-       struct talloc_chunk *tc = talloc_chunk_from_ptr(ptr);
-       tc->name = talloc_vasprintf(ptr, fmt, ap);
-       if (tc->name) {
-               talloc_set_name_const(tc->name, ".name");
-       }
-}
-
-/*
-  add a name to an existing pointer
-*/
-void talloc_set_name(const void *ptr, const char *fmt, ...)
-{
-       va_list ap;
-       va_start(ap, fmt);
-       talloc_set_name_v(ptr, fmt, ap);
-       va_end(ap);
-}
-
-/*
-   more efficient way to add a name to a pointer - the name must point to a 
-   true string constant
-*/
-void talloc_set_name_const(const void *ptr, const char *name)
-{
-       struct talloc_chunk *tc = talloc_chunk_from_ptr(ptr);
-       tc->name = name;
-}
-
-/*
-  create a named talloc pointer. Any talloc pointer can be named, and
-  talloc_named() operates just like talloc() except that it allows you
-  to name the pointer.
-*/
-void *talloc_named(const void *context, size_t size, const char *fmt, ...)
-{
-       va_list ap;
-       void *ptr;
-
-       ptr = _talloc(context, size);
-       if (ptr == NULL) return NULL;
-
-       va_start(ap, fmt);
-       talloc_set_name_v(ptr, fmt, ap);
-       va_end(ap);
-
-       return ptr;
-}
-
-/*
-  create a named talloc pointer. Any talloc pointer can be named, and
-  talloc_named() operates just like talloc() except that it allows you
-  to name the pointer.
-*/
-void *talloc_named_const(const void *context, size_t size, const char *name)
-{
-       void *ptr;
-
-       ptr = _talloc(context, size);
-       if (ptr == NULL) {
-               return NULL;
-       }
-
-       talloc_set_name_const(ptr, name);
-
-       return ptr;
-}
-
-/*
-  return the name of a talloc ptr, or "UNNAMED"
-*/
-const char *talloc_get_name(const void *ptr)
-{
-       struct talloc_chunk *tc = talloc_chunk_from_ptr(ptr);
-       if (tc->name == TALLOC_MAGIC_REFERENCE) {
-               return ".reference";
-       }
-       if (tc->name) {
-               return tc->name;
-       }
-       return "UNNAMED";
-}
-
-/*
-  this is for compatibility with older versions of talloc
-*/
-void *talloc_init(const char *fmt, ...)
-{
-       va_list ap;
-       void *ptr;
-
-       ptr = _talloc(NULL, 0);
-       if (ptr == NULL) return NULL;
-
-       va_start(ap, fmt);
-       talloc_set_name_v(ptr, fmt, ap);
-       va_end(ap);
-
-       return ptr;
-}
-
-
-/* 
-   free a talloc pointer. This also frees all child pointers of this 
-   pointer recursively
-
-   return 0 if the memory is actually freed, otherwise -1. The memory
-   will not be freed if the ref_count is > 1 or the destructor (if
-   any) returns non-zero
-*/
-int talloc_free(void *ptr)
-{
-       struct talloc_chunk *tc;
-
-       if (ptr == NULL) {
-               return -1;
-       }
-
-       tc = talloc_chunk_from_ptr(ptr);
-
-       if (tc->refs) {
-               talloc_reference_destructor(tc->refs);
-               return -1;
-       }
-
-       if (tc->destructor) {
-               talloc_destructor_t d = tc->destructor;
-               if (d == (talloc_destructor_t)-1) {
-                       return -1;
-               }
-               tc->destructor = (talloc_destructor_t)-1;
-               if (d(ptr) == -1) {
-                       tc->destructor = d;
-                       return -1;
-               }
-               tc->destructor = NULL;
-       }
-
-       while (tc->child) {
-               /* we need to work out who will own an abandoned child
-                  if it cannot be freed. In priority order, the first
-                  choice is owner of any remaining reference to this
-                  pointer, the second choice is our parent, and the
-                  final choice is the null context. */
-               void *child = tc->child+1;
-               const void *new_parent = null_context;
-               if (tc->child->refs) {
-                       struct talloc_chunk *p = talloc_parent_chunk(tc->child->refs);
-                       if (p) new_parent = p+1;
-               }
-               if (talloc_free(child) == -1) {
-                       if (new_parent == null_context) {
-                               struct talloc_chunk *p = talloc_parent_chunk(ptr);
-                               if (p) new_parent = p+1;
-                       }
-                       talloc_steal(new_parent, child);
-               }
-       }
-
-       if (tc->parent) {
-               _TLIST_REMOVE(tc->parent->child, tc);
-               if (tc->parent->child) {
-                       tc->parent->child->parent = tc->parent;
-               }
-       } else {
-               if (tc->prev) tc->prev->next = tc->next;
-               if (tc->next) tc->next->prev = tc->prev;
-       }
-
-       tc->magic = TALLOC_MAGIC_FREE;
-
-       free(tc);
-       return 0;
-}
-
-
-
-/*
-  A talloc version of realloc. The context argument is only used if
-  ptr is NULL
-*/
-void *_talloc_realloc(const void *context, void *ptr, size_t size, const char *name)
-{
-       struct talloc_chunk *tc;
-       void *new_ptr;
-
-       /* size zero is equivalent to free() */
-       if (size == 0) {
-               talloc_free(ptr);
-               return NULL;
-       }
-
-       if (size >= MAX_TALLOC_SIZE) {
-               return NULL;
-       }
-
-       /* realloc(NULL) is equavalent to malloc() */
-       if (ptr == NULL) {
-               return talloc_named_const(context, size, name);
-       }
-
-       tc = talloc_chunk_from_ptr(ptr);
-
-       /* don't allow realloc on referenced pointers */
-       if (tc->refs) {
-               return NULL;
-       }
-
-       /* by resetting magic we catch users of the old memory */
-       tc->magic = TALLOC_MAGIC_FREE;
-
-#if ALWAYS_REALLOC
-       new_ptr = malloc(size + sizeof(*tc));
-       if (new_ptr) {
-               memcpy(new_ptr, tc, tc->size + sizeof(*tc));
-               free(tc);
-       }
-#else
-       new_ptr = realloc(tc, size + sizeof(*tc));
-#endif
-       if (!new_ptr) { 
-               tc->magic = TALLOC_MAGIC; 
-               return NULL; 
-       }
-
-       tc = new_ptr;
-       tc->magic = TALLOC_MAGIC;
-       if (tc->parent) {
-               tc->parent->child = new_ptr;
-       }
-
-       if (tc->prev) {
-               tc->prev->next = tc;
-       }
-       if (tc->next) {
-               tc->next->prev = tc;
-       }
-
-       tc->size = size;
-       talloc_set_name_const(tc+1, name);
-
-       return (void *)(tc+1);
-}
-
-/* 
-   move a lump of memory from one talloc context to another return the
-   ptr on success, or NULL if it could not be transferred
-*/
-void *talloc_steal(const void *new_ctx, const void *ptr)
-{
-       struct talloc_chunk *tc, *new_tc;
-
-       if (!ptr) {
-               return NULL;
-       }
-
-       if (new_ctx == NULL) {
-               new_ctx = null_context;
-       }
-
-       tc = talloc_chunk_from_ptr(ptr);
-
-       if (new_ctx == NULL) {
-               if (tc->parent) {
-                       _TLIST_REMOVE(tc->parent->child, tc);
-                       if (tc->parent->child) {
-                               tc->parent->child->parent = tc->parent;
-                       }
-               } else {
-                       if (tc->prev) tc->prev->next = tc->next;
-                       if (tc->next) tc->next->prev = tc->prev;
-               }
-               
-               tc->parent = tc->next = tc->prev = NULL;
-               return discard_const_p(void, ptr);
-       }
-
-       new_tc = talloc_chunk_from_ptr(new_ctx);
-
-       if (tc == new_tc) {
-               return discard_const_p(void, ptr);
-       }
-
-       if (tc->parent) {
-               _TLIST_REMOVE(tc->parent->child, tc);
-               if (tc->parent->child) {
-                       tc->parent->child->parent = tc->parent;
-               }
-       } else {
-               if (tc->prev) tc->prev->next = tc->next;
-               if (tc->next) tc->next->prev = tc->prev;
-       }
-
-       tc->parent = new_tc;
-       if (new_tc->child) new_tc->child->parent = NULL;
-       _TLIST_ADD(new_tc->child, tc);
-
-       return discard_const_p(void, ptr);
-}
-
-/*
-  return the total size of a talloc pool (subtree)
-*/
-off_t talloc_total_size(const void *ptr)
-{
-       off_t total = 0;
-       struct talloc_chunk *c, *tc;
-       
-       if (ptr == NULL) {
-               ptr = null_context;
-       }
-       if (ptr == NULL) {
-               return 0;
-       }
-
-       tc = talloc_chunk_from_ptr(ptr);
-
-       total = tc->size;
-       for (c=tc->child;c;c=c->next) {
-               total += talloc_total_size(c+1);
-       }
-       return total;
-}
-
-/*
-  return the total number of blocks in a talloc pool (subtree)
-*/
-off_t talloc_total_blocks(const void *ptr)
-{
-       off_t total = 0;
-       struct talloc_chunk *c, *tc = talloc_chunk_from_ptr(ptr);
-
-       total++;
-       for (c=tc->child;c;c=c->next) {
-               total += talloc_total_blocks(c+1);
-       }
-       return total;
-}
-
-/*
-  return the number of external references to a pointer
-*/
-static int talloc_reference_count(const void *ptr)
-{
-       struct talloc_chunk *tc = talloc_chunk_from_ptr(ptr);
-       struct talloc_reference_handle *h;
-       int ret = 0;
-
-       for (h=tc->refs;h;h=h->next) {
-               ret++;
-       }
-       return ret;
-}
-
-/*
-  report on memory usage by all children of a pointer, giving a full tree view
-*/
-static void talloc_report_depth(const void *ptr, FILE *f, int depth)
-{
-       struct talloc_chunk *c, *tc = talloc_chunk_from_ptr(ptr);
-
-       for (c=tc->child;c;c=c->next) {
-               if (c->name == TALLOC_MAGIC_REFERENCE) {
-                       struct talloc_reference_handle *handle = (void *)(c+1);
-                       const char *name2 = talloc_get_name(handle->ptr);
-                       fprintf(f, "%*sreference to: %s\n", depth*4, "", name2);
-               } else {
-                       const char *name = talloc_get_name(c+1);
-                       fprintf(f, "%*s%-30s contains %6lu bytes in %3lu blocks (ref %d)\n", 
-                               depth*4, "",
-                               name,
-                               (unsigned long)talloc_total_size(c+1),
-                               (unsigned long)talloc_total_blocks(c+1),
-                               talloc_reference_count(c+1));
-                       talloc_report_depth(c+1, f, depth+1);
-               }
-       }
-
-}
-
-/*
-  report on memory usage by all children of a pointer, giving a full tree view
-*/
-void talloc_report_full(const void *ptr, FILE *f)
-{
-       if (ptr == NULL) {
-               ptr = null_context;
-       }
-       if (ptr == NULL) return;
-
-       fprintf(f,"full talloc report on '%s' (total %lu bytes in %lu blocks)\n", 
-               talloc_get_name(ptr), 
-               (unsigned long)talloc_total_size(ptr),
-               (unsigned long)talloc_total_blocks(ptr));
-
-       talloc_report_depth(ptr, f, 1);
-}
-
-/*
-  report on memory usage by all children of a pointer
-*/
-void talloc_report(const void *ptr, FILE *f)
-{
-       struct talloc_chunk *c, *tc;
-
-       if (ptr == NULL) {
-               ptr = null_context;
-       }
-       if (ptr == NULL) return;
-       
-       fprintf(f,"talloc report on '%s' (total %lu bytes in %lu blocks)\n", 
-               talloc_get_name(ptr), 
-               (unsigned long)talloc_total_size(ptr),
-               (unsigned long)talloc_total_blocks(ptr));
-
-       tc = talloc_chunk_from_ptr(ptr);
-
-       for (c=tc->child;c;c=c->next) {
-               fprintf(f, "\t%-30s contains %6lu bytes in %3lu blocks\n", 
-                       talloc_get_name(c+1),
-                       (unsigned long)talloc_total_size(c+1),
-                       (unsigned long)talloc_total_blocks(c+1));
-       }
-
-}
-
-/*
-  report on any memory hanging off the null context
-*/
-static void talloc_report_null(void)
-{
-       if (talloc_total_size(null_context) != 0) {
-               talloc_report(null_context, stderr);
-       }
-}
-
-/*
-  report on any memory hanging off the null context
-*/
-static void talloc_report_null_full(void)
-{
-       if (talloc_total_size(null_context) != 0) {
-               talloc_report_full(null_context, stderr);
-       }
-}
-
-/*
-  enable leak reporting on exit
-*/
-void talloc_enable_leak_report(void)
-{
-       null_context = talloc_named_const(NULL, 0, "null_context");
-       atexit(talloc_report_null);
-}
-
-/*
-  enable full leak reporting on exit
-*/
-void talloc_enable_leak_report_full(void)
-{
-       null_context = talloc_named_const(NULL, 0, "null_context");
-       atexit(talloc_report_null_full);
-}
-
-/* 
-   talloc and zero memory. 
-*/
-void *_talloc_zero(const void *ctx, size_t size, const char *name)
-{
-       void *p = talloc_named_const(ctx, size, name);
-
-       if (p) {
-               memset(p, '\0', size);
-       }
-
-       return p;
-}
-
-
-/*
-  memdup with a talloc. 
-*/
-void *_talloc_memdup(const void *t, const void *p, size_t size, const char *name)
-{
-       void *newp = talloc_named_const(t, size, name);
-
-       if (newp) {
-               memcpy(newp, p, size);
-       }
-
-       return newp;
-}
-
-/*
-  strdup with a talloc 
-*/
-char *talloc_strdup(const void *t, const char *p)
-{
-       char *ret;
-       if (!p) {
-               return NULL;
-       }
-       ret = talloc_memdup(t, p, strlen(p) + 1);
-       if (ret) {
-               talloc_set_name_const(ret, ret);
-       }
-       return ret;
-}
-
-/*
-  strndup with a talloc 
-*/
-char *talloc_strndup(const void *t, const char *p, size_t n)
-{
-       size_t len;
-       char *ret;
-
-       for (len=0; p[len] && len<n; len++) ;
-
-       ret = talloc(t, len + 1);
-       if (!ret) { return NULL; }
-       memcpy(ret, p, len);
-       ret[len] = 0;
-       return ret;
-}
-
-#ifndef VA_COPY
-#ifdef HAVE_VA_COPY
-#define VA_COPY(dest, src) __va_copy(dest, src)
-#else
-#define VA_COPY(dest, src) (dest) = (src)
-#endif
-#endif
-
-char *talloc_vasprintf(const void *t, const char *fmt, va_list ap)
-{      
-       int len;
-       char *ret;
-       va_list ap2;
-       
-       VA_COPY(ap2, ap);
-
-       len = vsnprintf(NULL, 0, fmt, ap2);
-
-       ret = talloc(t, len+1);
-       if (ret) {
-               VA_COPY(ap2, ap);
-               vsnprintf(ret, len+1, fmt, ap2);
-               talloc_set_name_const(ret, ret);
-       }
-
-       return ret;
-}
-
-
-/*
-  Perform string formatting, and return a pointer to newly allocated
-  memory holding the result, inside a memory pool.
- */
-char *talloc_asprintf(const void *t, const char *fmt, ...)
-{
-       va_list ap;
-       char *ret;
-
-       va_start(ap, fmt);
-       ret = talloc_vasprintf(t, fmt, ap);
-       va_end(ap);
-       return ret;
-}
-
-
-/**
- * Realloc @p s to append the formatted result of @p fmt and @p ap,
- * and return @p s, which may have moved.  Good for gradually
- * accumulating output into a string buffer.
- **/
-
-static char *talloc_vasprintf_append(char *s, const char *fmt, va_list ap) PRINTF_ATTRIBUTE(2,0);
-
-static char *talloc_vasprintf_append(char *s, const char *fmt, va_list ap)
-{      
-       int len, s_len;
-       va_list ap2;
-
-       VA_COPY(ap2, ap);
-
-       if (s) {
-               s_len = strlen(s);
-       } else {
-               s_len = 0;
-       }
-       len = vsnprintf(NULL, 0, fmt, ap2);
-
-       s = talloc_realloc(NULL, s, s_len + len+1);
-       if (!s) return NULL;
-
-       VA_COPY(ap2, ap);
-
-       vsnprintf(s+s_len, len+1, fmt, ap2);
-       talloc_set_name_const(s, s);
-
-       return s;
-}
-
-/*
-  Realloc @p s to append the formatted result of @p fmt and return @p
-  s, which may have moved.  Good for gradually accumulating output
-  into a string buffer.
- */
-char *talloc_asprintf_append(char *s, const char *fmt, ...)
-{
-       va_list ap;
-
-       va_start(ap, fmt);
-       s = talloc_vasprintf_append(s, fmt, ap);
-       va_end(ap);
-       return s;
-}
-
-/*
-  alloc an array, checking for integer overflow in the array size
-*/
-void *talloc_array(const void *ctx, size_t el_size, unsigned count, const char *name)
-{
-       if (count >= MAX_TALLOC_SIZE/el_size) {
-               return NULL;
-       }
-       return talloc_named_const(ctx, el_size * count, name);
-}
-
-/*
-  alloc an zero array, checking for integer overflow in the array size
-*/
-void *talloc_zero_array(const void *ctx, size_t el_size, unsigned count, const char *name)
-{
-       if (count >= MAX_TALLOC_SIZE/el_size) {
-               return NULL;
-       }
-       return _talloc_zero(ctx, el_size * count, name);
-}
-
-
-/*
-  realloc an array, checking for integer overflow in the array size
-*/
-void *talloc_realloc_array(const void *ctx, void *ptr, size_t el_size, unsigned count, const char *name)
-{
-       if (count >= MAX_TALLOC_SIZE/el_size) {
-               return NULL;
-       }
-       ptr = talloc_realloc(ctx, ptr, el_size * count);
-       if (ptr) {
-               talloc_set_name_const(ptr, name);
-       }
-       return ptr;
-}
-
-/*
-  a function version of talloc_realloc(), so it can be passed as a function pointer
-  to libraries that want a realloc function (a realloc function encapsulates
-  all the basic capabilities of an allocation library, which is why this is useful)
-*/
-void *talloc_realloc_fn(const void *context, void *ptr, size_t size)
-{
-       return _talloc_realloc(context, ptr, size, NULL);
-}
diff --git a/source4/lib/talloc/Makefile.talloc b/source4/lib/talloc/Makefile.talloc
new file mode 100644 (file)
index 0000000..88cb0e7
--- /dev/null
@@ -0,0 +1,16 @@
+OPT = -g -ftest-coverage -fprofile-arcs 
+LIBS = -lgcov
+#OPT = -g
+
+CFLAGS = $(OPT) -Wall
+
+all: testsuite
+
+testsuite: talloc.o testsuite.o
+       $(CC) $(CFLAGS) -o testsuite testsuite.o talloc.o $(LIBS)
+
+clean:
+       rm -f *~ *.o testsuite *.gc??
+
+gcov:
+       gcov talloc.c
diff --git a/source4/lib/talloc/config.m4 b/source4/lib/talloc/config.m4
new file mode 100644 (file)
index 0000000..298c669
--- /dev/null
@@ -0,0 +1,3 @@
+if test x"$experimental" = x"yes"; then
+       SMB_LIBRARY_ENABLE(libtalloc,YES)
+fi
diff --git a/source4/lib/talloc/config.mk b/source4/lib/talloc/config.mk
new file mode 100644 (file)
index 0000000..2f38fe0
--- /dev/null
@@ -0,0 +1,23 @@
+################################################
+# Start SUBSYSTEM LIBTALLOC
+[SUBSYSTEM::LIBTALLOC]
+INIT_OBJ_FILES = \
+               lib/talloc/talloc.o
+REQUIRED_SUBSYSTEMS = \
+               LIBREPLACE
+#
+# End SUBSYSTEM LIBTALLOC
+################################################
+
+################################################
+# Start LIBRARY LIBTALLOC
+[LIBRARY::libtalloc]
+MAJOR_VERSION = 0
+MINOR_VERSION = 0
+RELEASE_VERSION = 1
+REQUIRED_SUBSYSTEMS = \
+               LIBTALLOC
+#
+# End LIBRARY LIBTALLOC
+################################################
+
similarity index 99%
rename from source4/lib/talloc.c
rename to source4/lib/talloc/talloc.c
index ce1a325ebd72364efd4ca524fa41efc0363ec38a..4666e28288ca914c672f93c2acfe7356d4eade51 100644 (file)
@@ -564,6 +564,9 @@ void *_talloc_realloc(const void *context, void *ptr, size_t size, const char *n
        if (tc->parent) {
                tc->parent->child = new_ptr;
        }
+       if (tc->child) {
+               tc->child->parent = new_ptr;
+       }
 
        if (tc->prev) {
                tc->prev->next = tc;
@@ -857,6 +860,7 @@ char *talloc_strndup(const void *t, const char *p, size_t n)
        if (!ret) { return NULL; }
        memcpy(ret, p, len);
        ret[len] = 0;
+       talloc_set_name_const(ret, ret);
        return ret;
 }
 
similarity index 98%
rename from source4/include/talloc.h
rename to source4/lib/talloc/talloc.h
index ffb4c9f2528030c2ce375c8fb4286fcddd7b9864..6ebba447aad97ad04c1d5690dcdcb6a6247d18fd 100644 (file)
@@ -90,7 +90,7 @@ char *talloc_asprintf_append(char *s,
 void *talloc_array(const void *ctx, size_t el_size, unsigned count, const char *name);
 void *talloc_zero_array(const void *ctx, size_t el_size, unsigned count, const char *name);
 void *talloc_realloc_array(const void *ctx, void *ptr, size_t el_size, unsigned count, const char *name);
-void *talloc_ldb_alloc(void *context, void *ptr, size_t size);
+void *talloc_realloc_fn(const void *context, void *ptr, size_t size);
 
 #endif
 
similarity index 91%
rename from source4/torture/local/talloc.c
rename to source4/lib/talloc/testsuite.c
index 69becf67ccd3663bf6a2b0773bf2bfcba63aa13a..22444b11168b632bbf4aaa23d0afd89e8071b2d7 100644 (file)
 #ifndef _SAMBA_BUILD_
 typedef enum {False=0,True=1} BOOL;
 
-static struct timeval current_time(void)
+static struct timeval timeval_current(void)
 {
        struct timeval tv;
-       GetTimeOfDay(&tv);
+       gettimeofday(&tv, NULL);
        return tv;
 }
 
-static double elapsed_time(struct timeval *tv)
+static double timeval_elapsed(struct timeval *tv)
 {
-       struct timeval tv2 = current_time();
+       struct timeval tv2 = timeval_current();
        return (tv2.tv_sec - tv->tv_sec) + 
               (tv2.tv_usec - tv->tv_usec)*1.0e-6;
 }
@@ -564,6 +564,40 @@ static BOOL test_realloc(void)
        return True;
 }
 
+
+/*
+  test realloc with a child
+*/
+static BOOL test_realloc_child(void)
+{
+       void *root;
+       struct el1 {
+               int count;
+               struct el2 {
+                       const char *name;
+               } **list;
+       } *el1;
+       struct el2 *el2;
+
+       printf("TESTING REALLOC WITH CHILD\n");
+
+       root = talloc(NULL, 0);
+
+       el1 = talloc_p(root, struct el1);
+       el1->list = talloc_p(el1, struct el2 *);
+       el1->list[0] = talloc_p(el1->list, struct el2);
+       el1->list[0]->name = talloc_strdup(el1->list[0], "testing");
+       
+       el2 = talloc_p(el1->list, struct el2);
+
+       el1->list = talloc_realloc_p(el1, el1->list, struct el2 *, 2);
+       el1->list[1] = el2;
+
+       talloc_free(root);
+
+       return True;
+}
+
 /*
   test steal
 */
@@ -645,6 +679,31 @@ static BOOL test_ldb(void)
        return True;
 }
 
+
+static BOOL test_unref_reparent(void)
+{
+       void *root, *p1, *p2, *c1;
+
+       printf("TESTING UNREFERENCE AFTER PARENT FREED\n");
+
+       root = talloc_named_const(NULL, 0, "root");
+       p1 = talloc_named_const(root, 1, "orig parent");
+       p2 = talloc_named_const(root, 1, "parent by reference");
+
+       c1 = talloc_named_const(p1, 1, "child");
+       talloc_reference(p2, c1);
+
+       talloc_free(p1);
+       talloc_unlink(p2, c1);
+
+       CHECK_SIZE(root, 1);
+
+       talloc_free(p2);
+       talloc_free(root);
+
+       return True;
+}
+
 /*
   measure the speed of talloc versus malloc
 */
@@ -701,7 +760,9 @@ BOOL torture_local_talloc(void)
        ret &= test_unlink1();
        ret &= test_misc();
        ret &= test_realloc();
+       ret &= test_realloc_child();
        ret &= test_steal();
+       ret &= test_unref_reparent();
        ret &= test_ldb();
        if (ret) {
                ret &= test_speed();
@@ -713,9 +774,9 @@ BOOL torture_local_talloc(void)
 
 
 #ifndef _SAMBA_BUILD_
-int main(void)
+ int main(void)
 {
-       if (!torture_local_talloc(0)) {
+       if (!torture_local_talloc()) {
                printf("ERROR: TESTSUIE FAILED\n");
                return -1;
        }
index 7c7b6379db67575c584beace723a4ca1765e081f..658d1f1ba6be365f545eb0e031686f2fd2433295 100644 (file)
@@ -133,7 +133,7 @@ REQUIRED_SUBSYSTEMS = \
 [SUBSYSTEM::TORTURE_LOCAL]
 ADD_OBJ_FILES = \
                torture/local/iconv.o \
-               torture/local/talloc.o \
+               lib/talloc/testsuite.o \
                torture/local/messaging.o \
                torture/local/binding_string.o \
                torture/local/idtree.o