aboutsummaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/mem.c248
-rw-r--r--src/mem.h64
2 files changed, 312 insertions, 0 deletions
diff --git a/src/mem.c b/src/mem.c
new file mode 100644
index 0000000..4b1a467
--- /dev/null
+++ b/src/mem.c
@@ -0,0 +1,248 @@
+/* $calcurse: mem.c,v 1.1 2008/12/28 13:15:18 culot Exp $ */
+
+/*
+ * Calcurse - text-based organizer
+ * Copyright (c) 2008 Frederic Culot
+ *
+ * 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., 59 Temple Place - Suite 330,
+ * Boston, MA 02111-1307, USA.
+ *
+ * Send your feedback or comments to : calcurse@culot.org
+ * Calcurse home page : http://culot.org/calcurse
+ *
+ */
+
+#include <string.h>
+#include <stdio.h>
+
+#include "i18n.h"
+#include "utils.h"
+#include "mem.h"
+
+enum {
+ BLK_STATE,
+ BLK_SIZE,
+ BLK_ID,
+ EXTRA_SPACE_START
+};
+
+#define EXTRA_SPACE_END 1
+#define EXTRA_SPACE EXTRA_SPACE_START + EXTRA_SPACE_END
+
+#define MAGIC_ALLOC 0xda
+#define MAGIC_FREE 0xdf
+
+struct mem_blk_s {
+ unsigned id, size;
+ const char *pos;
+ struct mem_blk_s *next;
+};
+
+typedef struct {
+ unsigned ncall, nalloc, nfree;
+ struct mem_blk_s *blk;
+} mem_stats_t;
+
+static mem_stats_t mstats;
+
+
+unsigned
+stats_add_blk (size_t size, const char *pos)
+{
+ struct mem_blk_s *o, **i;
+
+ o = malloc (sizeof (struct mem_blk_s));
+ EXIT_IF (o == 0, _("could not allocate memory to store block info"));
+
+ mstats.ncall++;
+
+ o->pos = pos;
+ o->size = (unsigned)size;
+ o->next = 0;
+
+ i = &mstats.blk;
+ for (i = &mstats.blk; *i; i = &(*i)->next)
+ ;
+ o->id = mstats.ncall;
+ *i = o;
+
+ return o->id;
+}
+
+void
+stats_del_blk (unsigned id)
+{
+ struct mem_blk_s *o, **i;
+
+ EXIT_IF (id < 0, _("Incorrect block id"));
+
+ i = &mstats.blk;
+ for (o = mstats.blk; o; o = o->next)
+ {
+ if (o->id == id)
+ {
+ *i = o->next;
+ free (o);
+ return;
+ }
+ i = &o->next;
+ }
+
+ EXIT (_("Block not found"));
+ /* NOTREACHED */
+}
+
+void *
+dbg_malloc (size_t size, const char *pos)
+{
+ unsigned *buf;
+
+ if (size == 0)
+ return (void *)0;
+
+ size = EXTRA_SPACE + (size + sizeof (unsigned) - 1) / sizeof (unsigned);
+ if ((buf = (unsigned *)malloc (size * sizeof (unsigned))) == 0)
+ return (void *)0;
+
+ buf[BLK_STATE] = MAGIC_ALLOC; /* state of the block */
+ buf[BLK_SIZE] = size; /* size of the block */
+ buf[BLK_ID] = stats_add_blk (size, pos); /* identify a block by its id */
+ buf[size - 1] = buf[BLK_ID]; /* mark at end of block */
+
+ mstats.nalloc += size;
+
+ return (void *)(buf + EXTRA_SPACE_START);
+}
+
+void *
+dbg_calloc (size_t nmemb, size_t size, const char *pos)
+{
+ void *buf;
+
+ if (!nmemb || !size)
+ return (void *)0;
+
+ EXIT_IF (nmemb > SIZE_MAX / size, _("overflow at %s"), pos);
+
+ size *= nmemb;
+ if ((buf = dbg_malloc (size, pos)) == 0)
+ return (void *)0;
+
+ bzero (buf, size);
+
+ return buf;
+}
+
+void *
+dbg_realloc (void *ptr, size_t size, const char *pos)
+{
+ unsigned *buf, old_size;
+
+ if (size == 0 || ptr == 0)
+ return (void *)0;
+
+ if ((buf = dbg_malloc (size, pos)) == 0)
+ return (void *)0;
+
+ old_size = *((unsigned *)ptr - EXTRA_SPACE_START + BLK_SIZE);
+ bcopy (ptr, buf + EXTRA_SPACE_START, old_size);
+ mem_free (ptr);
+
+ mstats.nalloc += size;
+
+ return (void *)(buf + EXTRA_SPACE_START);
+}
+
+char *
+dbg_strdup (const char *s, const char *pos)
+{
+ size_t size;
+ char *buf;
+
+ if (s == 0)
+ return (char *)0;
+
+ size = strlen (s);
+ if ((buf = dbg_malloc (size + 1, pos)) == 0)
+ return (char *)0;
+
+ return strncpy (buf, s, size + 1);
+}
+
+void
+dbg_free (void *ptr, const char *pos)
+{
+ unsigned *buf, size;
+
+ if (ptr == 0)
+ return;
+
+ buf = (unsigned *)ptr - EXTRA_SPACE_START;
+ size = buf[BLK_SIZE];
+
+ EXIT_IF (buf[BLK_STATE] == MAGIC_FREE,
+ _("block seems already freed at %s"), pos);
+ EXIT_IF (buf[BLK_STATE] != MAGIC_ALLOC,
+ _("corrupt block header at %s"), pos);
+ EXIT_IF (buf[size - 1] != buf[BLK_ID],
+ _("corrupt block end at %s, (end = %u, should be %d)"), pos,
+ buf[size - 1], buf[BLK_ID]);
+
+ buf[0] = MAGIC_FREE;
+
+ stats_del_blk (buf[BLK_ID]);
+
+ free (buf);
+ mstats.nfree += size;
+}
+
+
+#ifdef CALCURSE_MEMORY_DEBUG
+
+static void
+dump_block_info (struct mem_blk_s *blk)
+{
+ if (blk == 0)
+ return;
+
+ printf (_("---==== MEMORY BLOCK ====----------------\n"));
+ printf (_(" id: %u\n"), blk->id);
+ printf (_(" size: %u\n"), blk->size);
+ printf (_(" allocated in: %s\n"), blk->pos);
+ printf (_("-----------------------------------------\n"));
+}
+
+void
+mem_stats (void)
+{
+ printf ("\n");
+ printf (_("+------------------------------+\n"));
+ printf (_("| calcurse memory usage report |\n"));
+ printf (_("+------------------------------+\n"));
+ printf (_(" number of calls: %u\n"), mstats.ncall);
+ printf (_(" allocated blocks: %u\n"), mstats.nalloc);
+ printf (_(" unfreed blocks: %u\n"), mstats.nalloc - mstats.nfree);
+ printf ("\n");
+
+ if (mstats.nfree < mstats.nalloc)
+ {
+ struct mem_blk_s *blk;
+
+ for (blk = mstats.blk; blk; blk = blk->next)
+ dump_block_info (blk);
+ }
+}
+
+#endif /* CALCURSE_MEMORY_DEBUG */
diff --git a/src/mem.h b/src/mem.h
new file mode 100644
index 0000000..703cbb0
--- /dev/null
+++ b/src/mem.h
@@ -0,0 +1,64 @@
+/* $calcurse: mem.h,v 1.1 2008/12/28 13:15:18 culot Exp $ */
+
+/*
+ * Calcurse - text-based organizer
+ * Copyright (c) 2008 Frederic Culot
+ *
+ * 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., 59 Temple Place - Suite 330,
+ * Boston, MA 02111-1307, USA.
+ *
+ * Send your feedback or comments to : calcurse@culot.org
+ * Calcurse home page : http://culot.org/calcurse
+ *
+ */
+
+#ifndef CALCURSE_MEM_H
+#define CALCURSE_MEM_H
+
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif /* HAVE_CONFIG_H */
+
+#ifdef CALCURSE_MEMORY_DEBUG
+
+#include <stdlib.h>
+
+#include "utils.h"
+
+#define mem_malloc(s) dbg_malloc ((s), __FILE_POS__)
+#define mem_calloc(n, s) dbg_calloc ((n), (s), __FILE_POS__)
+#define mem_realloc(p, s) dbg_realloc ((p), (s), __FILE_POS__)
+#define mem_strdup(s) dbg_strdup ((s), __FILE_POS__)
+#define mem_free(p) dbg_free ((p), __FILE_POS__)
+
+void *dbg_malloc (size_t, const char *);
+void *dbg_calloc (size_t, size_t, const char *);
+void *dbg_realloc (void *, size_t, const char *);
+char *dbg_strdup (const char *, const char *);
+void dbg_free (void *, const char *);
+void mem_stats (void);
+
+#else /* !CALCURSE_MEMORY_DEBUG */
+
+#define mem_malloc(s) malloc ((s))
+#define mem_calloc(n, s) calloc ((n), (s))
+#define mem_realloc(p, s) realloc ((p), (s))
+#define mem_strdup(s) strdup ((s))
+#define mem_free(p) free ((p))
+#define mem_stats()
+
+#endif /* CALCURSE_MEMORY_DEBUG */
+
+#endif /* CALCURSE_MEM_H */