]> git.ipfire.org Git - thirdparty/asterisk.git/commitdiff
add support for tracking thread-local-storage objects that exist via 'threadstorage...
authorKevin P. Fleming <kpfleming@digium.com>
Thu, 4 Jan 2007 22:51:01 +0000 (22:51 +0000)
committerKevin P. Fleming <kpfleming@digium.com>
Thu, 4 Jan 2007 22:51:01 +0000 (22:51 +0000)
git-svn-id: https://origsvn.digium.com/svn/asterisk/branches/1.4@49553 65c4cc65-6c06-0410-ace0-fbb531ad65f3

build_tools/cflags.xml
include/asterisk.h
include/asterisk/threadstorage.h
main/Makefile
main/asterisk.c
main/threadstorage.c [new file with mode: 0644]
main/utils.c

index 7c2fbddffb707e95388f7f3486e363caa1582b0e..994c96203e2e7f7e5b13373647de5b2ecf40e7cc 100644 (file)
@@ -5,6 +5,8 @@
                </member>
                <member name="DEBUG_THREADS" displayname="Enable Thread Debugging">
                </member>
+               <member name="DEBUG_THREADLOCALS" displayname="Enable Thread-Local-Storage Debugging">
+               </member>
                <member name="DETECT_DEADLOCKS" displayname="Detect Deadlocks">
                </member>
                <member name="DO_CRASH" displayname="Crash on fatal errors">
index 6579496f7a9fd2fe0bc09a16a43c649f48c20962..fcd47799ff99df2ee1651dd09279a58eb170281f 100644 (file)
@@ -71,6 +71,7 @@ void ast_builtins_init(void);                 /*!< Provided by cli.c */
 int dnsmgr_init(void);                         /*!< Provided by dnsmgr.c */ 
 void dnsmgr_start_refresh(void);               /*!< Provided by dnsmgr.c */
 int dnsmgr_reload(void);                       /*!< Provided by dnsmgr.c */
+void threadstorage_init(void);                 /*!< Provided by threadstorage.c */
 
 /* Many headers need 'ast_channel' to be defined */
 struct ast_channel;
index 14c6d65b94942cbf5be61a4ac55ae2a78a471aca..f17991957d95df7a7dd5106ce4efb6a6b7ac6841 100644 (file)
@@ -43,6 +43,13 @@ struct ast_threadstorage {
        void (*key_init)(void);
 };
 
+
+#if defined(DEBUG_THREADLOCALS)
+void __ast_threadstorage_object_add(void *key, size_t len, const char *file, const char *function, unsigned int line);
+void __ast_threadstorage_object_remove(void *key);
+void __ast_threadstorage_object_replace(void *key_old, void *key_new, size_t len);
+#endif /* defined(DEBUG_THREADLOCALS) */
+
 /*!
  * \brief Define a thread storage variable
  *
@@ -61,6 +68,7 @@ struct ast_threadstorage {
 #define AST_THREADSTORAGE(name, name_init) \
        AST_THREADSTORAGE_CUSTOM(name, name_init, ast_free) 
 
+#if !defined(DEBUG_THREADLOCALS)
 #define AST_THREADSTORAGE_CUSTOM(name, name_init, cleanup)  \
 static void name_init(void);                                \
 static struct ast_threadstorage name = {                    \
@@ -71,6 +79,23 @@ static void name_init(void)                                 \
 {                                                           \
        pthread_key_create(&(name).key, cleanup);           \
 }
+#else /* defined(DEBUG_THREADLOCALS) */
+#define AST_THREADSTORAGE_CUSTOM(name, name_init, cleanup)  \
+static void name_init(void);                                \
+static struct ast_threadstorage name = {                    \
+       .once = PTHREAD_ONCE_INIT,                          \
+       .key_init = name_init,                              \
+};                                                          \
+static void __cleanup_##name(void *data)                   \
+{                                                          \
+       __ast_threadstorage_object_remove(data);            \
+       cleanup(data);                                      \
+}                                                          \
+static void name_init(void)                                 \
+{                                                           \
+       pthread_key_create(&(name).key, __cleanup_##name);  \
+}
+#endif /* defined(DEBUG_THREADLOCALS) */
 
 /*!
  * \brief Retrieve thread storage
@@ -102,6 +127,7 @@ static void name_init(void)                                 \
  * }
  * \endcode
  */
+#if !defined(DEBUG_THREADLOCALS)
 AST_INLINE_API(
 void *ast_threadstorage_get(struct ast_threadstorage *ts, size_t init_size),
 {
@@ -117,6 +143,26 @@ void *ast_threadstorage_get(struct ast_threadstorage *ts, size_t init_size),
        return buf;
 }
 )
+#else /* defined(DEBUG_THREADLOCALS) */
+AST_INLINE_API(
+void *__ast_threadstorage_get(struct ast_threadstorage *ts, size_t init_size, const char *file, const char *function, unsigned int line),
+{
+       void *buf;
+
+       pthread_once(&ts->once, ts->key_init);
+       if (!(buf = pthread_getspecific(ts->key))) {
+               if (!(buf = ast_calloc(1, init_size)))
+                       return NULL;
+               pthread_setspecific(ts->key, buf);
+               __ast_threadstorage_object_add(buf, init_size, file, function, line);
+       }
+
+       return buf;
+}
+)
+
+#define ast_threadstorage_get(ts, init_size) __ast_threadstorage_get(ts, init_size, __FILE__, __PRETTY_FUNCTION__, __LINE__)
+#endif /* defined(DEBUG_THREADLOCALS) */
 
 /*!
  * \brief A dynamic length string
@@ -184,6 +230,7 @@ struct ast_dynamic_str * attribute_malloc ast_dynamic_str_create(size_t init_len
  * }
  * \endcode
  */
+#if !defined(DEBUG_THREADLOCALS)
 AST_INLINE_API(
 struct ast_dynamic_str *ast_dynamic_str_thread_get(struct ast_threadstorage *ts,
        size_t init_len),
@@ -199,6 +246,25 @@ struct ast_dynamic_str *ast_dynamic_str_thread_get(struct ast_threadstorage *ts,
        return buf;
 }
 )
+#else /* defined(DEBUG_THREADLOCALS) */
+AST_INLINE_API(
+struct ast_dynamic_str *__ast_dynamic_str_thread_get(struct ast_threadstorage *ts,
+       size_t init_len, const char *file, const char *function, unsigned int line),
+{
+       struct ast_dynamic_str *buf;
+
+       if (!(buf = __ast_threadstorage_get(ts, sizeof(*buf) + init_len, file, function, line)))
+               return NULL;
+       
+       if (!buf->len)
+               buf->len = init_len;
+
+       return buf;
+}
+)
+
+#define ast_dynamic_str_thread_get(ts, init_len) __ast_dynamic_str_thread_get(ts, init_len, __FILE__, __PRETTY_FUNCTION__, __LINE__)
+#endif /* defined(DEBUG_THREADLOCALS) */ 
 
 /*!
  * \brief Error codes from ast_dynamic_str_thread_build_va()
index 6f0b768bd4479cae25c07aaaf6c8d464da44da2b..d1fca754b4e9e8094b44f4d96a32f2b0b452544f 100644 (file)
@@ -26,7 +26,7 @@ OBJS= io.o sched.o logger.o frame.o loader.o config.o channel.o \
        utils.o plc.o jitterbuf.o dnsmgr.o devicestate.o \
        netsock.o slinfactory.o ast_expr2.o ast_expr2f.o \
        cryptostub.o sha1.o http.o fixedjitterbuf.o abstract_jb.o \
-       strcompat.o
+       strcompat.o threadstorage.o
 
 # we need to link in the objects statically, not as a library, because
 # otherwise modules will not have them available if none of the static
index 929a83b4789fae03972e014eb58c33ac9e51fa28..71addc1c01443ff2fd2a0b8638f85f76ef50b0dc 100644 (file)
@@ -2714,6 +2714,9 @@ int main(int argc, char *argv[])
                printf(term_quit());
                exit(1);
        }
+
+       threadstorage_init();
+
        if (load_modules(1)) {
                printf(term_quit());
                exit(1);
diff --git a/main/threadstorage.c b/main/threadstorage.c
new file mode 100644 (file)
index 0000000..b4d38c0
--- /dev/null
@@ -0,0 +1,227 @@
+/*
+ * Asterisk -- An open source telephony toolkit.
+ *
+ * Copyright (C) 1999 - 2005, Digium, Inc.
+ *
+ * Kevin P. Fleming <kpfleming@digium.com>
+ *
+ * See http://www.asterisk.org for more information about
+ * the Asterisk project. Please do not directly contact
+ * any of the maintainers of this project for assistance;
+ * the project provides a web site, mailing lists and IRC
+ * channels for your use.
+ *
+ * This program is free software, distributed under the terms of
+ * the GNU General Public License Version 2. See the LICENSE file
+ * at the top of the source tree.
+ */
+
+/*! \file
+ *
+ * \brief Debugging support for thread-local-storage objects
+ *
+ * \author Kevin P. Fleming <kpfleming@digium.com>
+ */
+
+#include "asterisk.h"
+
+#if defined(DEBUG_THREADLOCALS)
+
+ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include "asterisk/logger.h"
+#include "asterisk/strings.h"
+#include "asterisk/utils.h"
+#include "asterisk/threadstorage.h"
+#include "asterisk/linkedlists.h"
+#include "asterisk/cli.h"
+
+struct tls_object {
+       void *key;
+       size_t size;
+       const char *file;
+       const char *function;
+       unsigned int line;
+       pthread_t thread;
+       AST_LIST_ENTRY(tls_object) entry;
+};
+
+static AST_LIST_HEAD_STATIC(tls_objects, tls_object);
+
+void __ast_threadstorage_object_add(void *key, size_t len, const char *file, const char *function, unsigned int line)
+{
+       struct tls_object *to;
+
+       if (!(to = ast_calloc(sizeof(*to), 1)))
+               return;
+
+       to->key = key;
+       to->size = len;
+       to->file = file;
+       to->function = function;
+       to->line = line;
+       to->thread = pthread_self();
+
+       AST_LIST_LOCK(&tls_objects);
+       AST_LIST_INSERT_TAIL(&tls_objects, to, entry);
+       AST_LIST_UNLOCK(&tls_objects);
+}
+
+void __ast_threadstorage_object_remove(void *key)
+{
+       struct tls_object *to;
+
+       AST_LIST_LOCK(&tls_objects);
+       AST_LIST_TRAVERSE_SAFE_BEGIN(&tls_objects, to, entry) {
+               if (to->key == key) {
+                       AST_LIST_REMOVE_CURRENT(&tls_objects, entry);
+                       break;
+               }
+       }
+       AST_LIST_TRAVERSE_SAFE_END;
+       AST_LIST_UNLOCK(&tls_objects);
+       if (to)
+               free(to);
+}
+
+void __ast_threadstorage_object_replace(void *key_old, void *key_new, size_t len)
+{
+       struct tls_object *to;
+
+       AST_LIST_LOCK(&tls_objects);
+       AST_LIST_TRAVERSE_SAFE_BEGIN(&tls_objects, to, entry) {
+               if (to->key == key_old) {
+                       to->key = key_new;
+                       to->size = len;
+                       break;
+               }
+       }
+       AST_LIST_TRAVERSE_SAFE_END;
+       AST_LIST_UNLOCK(&tls_objects);
+}
+
+static int handle_show_allocations(int fd, int argc, char *argv[])
+{
+       char *fn = NULL;
+       size_t len = 0;
+       unsigned int count = 0;
+       struct tls_object *to;
+
+       if (argc > 3)
+               fn = argv[3];
+
+       AST_LIST_LOCK(&tls_objects);
+
+       AST_LIST_TRAVERSE(&tls_objects, to, entry) {
+               if (fn && strcasecmp(to->file, fn))
+                       continue;
+
+               ast_cli(fd, "%10d bytes allocated in %20s at line %5d of %25s (thread %p)\n",
+                       (int) to->size, to->function, to->line, to->file, (void *) to->thread);
+               len += to->size;
+               count++;
+       }
+
+       AST_LIST_UNLOCK(&tls_objects);
+
+       ast_cli(fd, "%10d bytes allocated in %d allocation%s\n", (int) len, count, count > 1 ? "s" : "");
+       
+       return RESULT_SUCCESS;
+}
+
+static int handle_show_summary(int fd, int argc, char *argv[])
+{
+       char *fn = NULL;
+       size_t len = 0;
+       unsigned int count = 0;
+       struct tls_object *to;
+       struct file {
+               const char *name;
+               size_t len;
+               unsigned int count;
+               AST_LIST_ENTRY(file) entry;
+       } *file;
+       AST_LIST_HEAD_NOLOCK_STATIC(file_summary, file);
+
+       if (argc > 3)
+               fn = argv[3];
+
+       AST_LIST_LOCK(&tls_objects);
+
+       AST_LIST_TRAVERSE(&tls_objects, to, entry) {
+               if (fn && strcasecmp(to->file, fn))
+                       continue;
+
+               AST_LIST_TRAVERSE(&file_summary, file, entry) {
+                       if ((!fn && (file->name == to->file)) || (fn && (file->name == to->function)))
+                               break;
+               }
+
+               if (!file) {
+                       file = alloca(sizeof(*file));
+                       memset(file, 0, sizeof(*file));
+                       file->name = fn ? to->function : to->file;
+                       AST_LIST_INSERT_TAIL(&file_summary, file, entry);
+               }
+
+               file->len += to->size;
+               file->count++;
+       }
+
+       AST_LIST_UNLOCK(&tls_objects);
+       
+       AST_LIST_TRAVERSE(&file_summary, file, entry) {
+               len += file->len;
+               count += file->count;
+               if (fn) {
+                       ast_cli(fd, "%10d bytes in %d allocation%ss in function %s\n",
+                               (int) file->len, file->count, file->count > 1 ? "s" : "", file->name);
+               } else {
+                       ast_cli(fd, "%10d bytes in %d allocation%s in file %s\n",
+                               (int) file->len, file->count, file->count > 1 ? "s" : "", file->name);
+               }
+       }
+
+       ast_cli(fd, "%10d bytes allocated in %d allocation%s\n", (int) len, count, count > 1 ? "s" : "");
+       
+       return RESULT_SUCCESS;
+}
+
+static struct ast_cli_entry cli[] = {
+       {
+               .cmda = { "threadstorage", "show", "allocations", NULL },
+               .handler = handle_show_allocations,
+               .summary = "Display outstanding thread local storage allocations",
+               .usage =
+               "Usage: threadstorage show allocations [<file>]\n"
+               "       Dumps a list of all thread-specific memory allocations,\n"
+               "optionally limited to those from a specific file\n",
+       },
+       {
+               .cmda = { "threadstorage", "show", "summary", NULL },
+               .handler = handle_show_summary,
+               .summary = "Summarize outstanding memory allocations",
+               .usage =
+               "Usage: threadstorage show summary [<file>]\n"
+               "       Summarizes thread-specific memory allocations by file, or optionally\n"
+               "by function, if a file is specified\n",
+       },
+};
+
+void threadstorage_init(void)
+{
+       ast_cli_register_multiple(cli, sizeof(cli) / sizeof(cli[0]));
+}
+
+#else /* !defined(DEBUG_THREADLOCALS) */
+
+void threadstorage_init(void)
+{
+}
+
+#endif /* !defined(DEBUG_THREADLOCALS) */
+
index ff226fc115eec1a2dabbdc0f50a678338276885f..88bd23c198b2ce47f13ac5ac0a3c4fc01c286a57 100644 (file)
@@ -962,6 +962,9 @@ int ast_dynamic_str_thread_build_va(struct ast_dynamic_str **buf, size_t max_len
 {
        int res;
        int offset = (append && (*buf)->len) ? strlen((*buf)->str) : 0;
+#if defined(DEBUG_THREADLOCALS)
+       struct ast_dynamic_str *old_buf = *buf;
+#endif /* defined(DEBUG_THREADLOCALS) */
 
        res = vsnprintf((*buf)->str + offset, (*buf)->len - offset, fmt, ap);
 
@@ -984,8 +987,12 @@ int ast_dynamic_str_thread_build_va(struct ast_dynamic_str **buf, size_t max_len
                if (append)
                        (*buf)->str[offset] = '\0';
 
-               if (ts)
+               if (ts) {
                        pthread_setspecific(ts->key, *buf);
+#if defined(DEBUG_THREADLOCALS)
+                       __ast_threadstorage_object_replace(old_buf, *buf, (*buf)->len + sizeof(*(*buf)));
+#endif /* defined(DEBUG_THREADLOCALS) */
+               }
 
                /* va_end() and va_start() must be done before calling
                 * vsnprintf() again. */