Rocco Rutte:

[apps/madmutt.git] / lib / mem.c

diff --git a/lib/mem.c b/lib/mem.c
new file mode 100644 (file)
index 0000000..ba40e98
--- /dev/null
+++ b/lib/mem.c
@@ -0,0 +1,76 @@
+/*
+ * This file is part of mutt-ng, see http://www.muttng.org/.
+ * It's licensed under the GNU General Public License,
+ * please see the file GPL in the top level source directory.
+ */
+
+#include <stdlib.h>
+
+#include "mem.h"
+#include "exit.h"
+#include "intl.h"
+
+void *_safe_calloc (size_t nmemb, size_t size, int line, const char* fname) {
+  void *p;
+
+  if (!nmemb || !size)
+    return NULL;
+
+  if (((size_t) - 1) / nmemb <= size) {
+    exit_fatal ("safe_calloc", _("Integer overflow -- can't allocate memory!"),
+                line, fname, 1);
+    return (NULL);
+  }
+
+  if (!(p = calloc (nmemb, size))) {
+    exit_fatal ("safe_calloc", _("Out of memory!"), line, fname, 1);
+    return (NULL);
+  }
+  return p;
+}
+
+void *_safe_malloc (size_t siz, int line, const char* fname) {
+  void *p;
+
+  if (siz == 0)
+    return 0;
+  if ((p = (void *) malloc (siz)) == 0) {       /* __MEM_CHECKED__ */
+    exit_fatal ("safe_malloc", _("Out of memory!"), line, fname, 1);
+    return (NULL);
+  }
+  return (p);
+}
+
+void _safe_realloc (void *ptr, size_t siz, int line, const char* fname) {
+  void *r;
+  void **p = (void **) ptr;
+
+  if (siz == 0) {
+    if (*p) {
+      free (*p);                /* __MEM_CHECKED__ */
+      *p = NULL;
+    }
+    return;
+  }
+
+  if (*p)
+    r = (void *) realloc (*p, siz);     /* __MEM_CHECKED__ */
+  else {
+    /* realloc(NULL, nbytes) doesn't seem to work under SunOS 4.1.x  --- __MEM_CHECKED__ */
+    r = (void *) malloc (siz);  /* __MEM_CHECKED__ */
+  }
+
+  if (!r)
+    exit_fatal ("safe_realloc", _("Out of memory!"), line, fname, 1);
+
+  *p = r;
+}
+
+void safe_free (void *ptr) {
+  void **p = (void **) ptr;
+
+  if (*p) {
+    free (*p);                  /* __MEM_CHECKED__ */
+    *p = 0;
+  }
+}