[srvx-commits] commit: Add redzone and statistics reporting to debug allocator.

[Michael P. <md...@tr...>]

Revision: srvx--devo--1.3--patch-5
Archive: sr...@sr...--2005-srvx
Creator: Michael Poole <md...@tr...>
Date: Thu Jan 20 19:48:35 EST 2005
Standard-date: 2005-01-21 00:48:35 GMT
Modified-files: ChangeLog src/alloc-srvx.c src/opserv.c
New-patches: sr...@sr...--2005-srvx/srvx--devo--1.3--patch-5
Summary: Add redzone and statistics reporting to debug allocator.
Keywords: 

src/alloc-srvx.c (redzone): New variable to detect buffer overwrites.
(alloc_count, alloc_size): New variables to track allocation stats.
(srvx_malloc, srvx_realloc, srvx_free): Use the above.

src/opserv.c (OSMSG_STATS_MEMORY): New translation string.
(cmd_stats_memory): New (conditional) command function.
(init_opserv): Conditionally provide cmd_stats_memory.

* added files

    {arch}/srvx/srvx--devo/srvx--devo--1.3/sr...@sr...--2005-srvx/patch-log/patch-5

* modified files

--- orig/ChangeLog
+++ mod/ChangeLog
@@ -2,6 +2,25 @@
 # arch-tag: aut...@sr...--2005-srvx/srvx--devo--1.3
 #
 
+2005-01-21 00:48:35 GMT	Michael Poole <md...@tr...>	patch-5
+
+    Summary:
+      Add redzone and statistics reporting to debug allocator.
+    Revision:
+      srvx--devo--1.3--patch-5
+
+    src/alloc-srvx.c (redzone): New variable to detect buffer overwrites.
+    (alloc_count, alloc_size): New variables to track allocation stats.
+    (srvx_malloc, srvx_realloc, srvx_free): Use the above.
+    
+    src/opserv.c (OSMSG_STATS_MEMORY): New translation string.
+    (cmd_stats_memory): New (conditional) command function.
+    (init_opserv): Conditionally provide cmd_stats_memory.
+
+    modified files:
+     ChangeLog src/alloc-srvx.c src/opserv.c
+
+
 2005-01-20 23:41:17 GMT	Michael Poole <md...@tr...>	patch-4
 
     Summary:


--- orig/src/alloc-srvx.c
+++ mod/src/alloc-srvx.c
@@ -22,6 +22,7 @@
 /* cookies for anybody who recognizes these bytes without help :) */
 #define ALLOC_MAGIC 0x1acf
 #define FREE_MAGIC  0xfc1d
+const char redzone[] = { '\x03', '\x47', '\x76', '\xc7' };
 
 struct alloc_header {
     unsigned int file_id : 8;
@@ -32,6 +33,7 @@
 
 static char file_id_map[256][32];
 static unsigned int file_ids_used;
+unsigned long alloc_count, alloc_size;
 
 static int
 file_id_cmp(const void *a_, const void *b_)
@@ -57,13 +59,16 @@
 {
     struct alloc_header *block;
 
-    block = malloc(sizeof(*block) + size);
+    block = malloc(sizeof(*block) + size + sizeof(redzone));
     assert(block != NULL);
     memset(block, 0, sizeof(*block) + size);
+    memcpy((char*)(block + 1) + size, redzone, sizeof(redzone));
     block->file_id = get_file_id(file);
     block->line = line;
     block->size = size;
     block->magic = ALLOC_MAGIC;
+    alloc_count++;
+    alloc_size += size;
     return block + 1;
 }
 
@@ -75,17 +80,21 @@
     if (ptr) {
         block = (struct alloc_header *)ptr - 1;
         assert(block->magic == ALLOC_MAGIC);
+        assert(0 == memcmp((char*)(block + 1) + block->size, redzone, sizeof(redzone)));
         if (block->size >= size)
             return block + 1;
     }
 
-    newblock = malloc(sizeof(*newblock) + size);
+    newblock = malloc(sizeof(*newblock) + size + sizeof(redzone));
     assert(newblock != NULL);
-    memset(newblock, 0, sizeof(*newblock) + size);
+    memset(newblock, 0, sizeof(*newblock) + size + sizeof(redzone));
+    memcpy((char*)(newblock + 1) + size, redzone, sizeof(redzone));
     newblock->file_id = get_file_id(file);
     newblock->line = line;
     newblock->size = size;
     newblock->magic = ALLOC_MAGIC;
+    alloc_count++;
+    alloc_size += size;
 
     if (ptr) {
         memcpy(newblock + 1, block + 1, block->size);
@@ -93,6 +102,8 @@
         memset(block, 0, size);
         block->magic = FREE_MAGIC;
         free(block);
+        alloc_count--;
+        alloc_size -= size - sizeof(*block);
     }
 
     return newblock + 1;
@@ -120,9 +131,12 @@
         return;
     block = (struct alloc_header *)ptr - 1;
     assert(block->magic == ALLOC_MAGIC);
+    assert(0 == memcmp((char*)(block + 1) + block->size, redzone, sizeof(redzone)));
     size = block->size + sizeof(*block);
     memset(block, 0, size);
     block->magic = FREE_MAGIC;
     free(block);
+    alloc_count--;
+    alloc_size -= size - sizeof(*block);
     (void)file; (void)line;
 }


--- orig/src/opserv.c
+++ mod/src/opserv.c
@@ -228,6 +228,7 @@
     { "OSMSG_WARN_LISTSTART", "Channel activity warnings:" },
     { "OSMSG_WARN_LISTENTRY", "%s (%s)" },
     { "OSMSG_WARN_LISTEND", "End of activity warning list." },
+    { "OSMSG_STATS_MEMORY", "%u allocations totalling %u bytes." },
     { "OSMSG_UPLINK_CONNECTING", "Establishing connection with %s (%s:%d)." },
     { "OSMSG_CURRENT_UPLINK", "$b%s$b is already the current uplink." },
     { "OSMSG_INVALID_UPLINK", "$b%s$b is not a valid uplink name." },
@@ -1578,6 +1579,14 @@
     return 1;
 }
 
+#if defined(WITH_MALLOC_SRVX)
+static MODCMD_FUNC(cmd_stats_memory) {
+    extern unsigned long alloc_count, alloc_size;
+    reply("OSMSG_STATS_MEMORY", alloc_count, alloc_size);
+    return 1;
+}
+#endif
+
 static MODCMD_FUNC(cmd_dump)
 {
     char linedup[MAXLEN], *original;
@@ -4101,6 +4110,9 @@
     opserv_define_func("STATS UPLINK", cmd_stats_uplink, 0, 0, 0);
     opserv_define_func("STATS UPTIME", cmd_stats_uptime, 0, 0, 0);
     opserv_define_func("STATS WARN", cmd_stats_warn, 0, 0, 0);
+#if defined(WITH_MALLOC_SRVX)
+    opserv_define_func("STATS MEMORY", cmd_stats_memory, 0, 0, 0);
+#endif
     opserv_define_func("TRACE", cmd_trace, 100, 0, 3);
     opserv_define_func("TRACE PRINT", NULL, 0, 0, 0);
     opserv_define_func("TRACE COUNT", NULL, 0, 0, 0);