Credit John Beck's fixes.

[~andy/fetchmail] / xmalloc.c

diff --git a/xmalloc.c b/xmalloc.c
index 7bdd5303e81a169180ed5fa75a02f1c42f829a26..c2ca4a66e932a32d99716d851914b13ca05da6f0 100644 (file)
--- a/xmalloc.c
+++ b/xmalloc.c
@@ -1,16 +1,20 @@
 /*
  * xmalloc.c -- allocate space or die 
  *
+ * Copyright 1998 by Eric S. Raymond.
  * For license terms, see the file COPYING in this directory.
  */
 
-#include <config.h>
+#include "config.h"
+#include <sys/types.h>
 #include <stdio.h>
+#include <errno.h>
 #include <string.h>
 #if defined(STDC_HEADERS)
 #include  <stdlib.h>
 #endif
 #include "fetchmail.h"
+#include "i18n.h"
 
 #if defined(HAVE_VOIDPOINTER)
 #define XMALLOCTYPE void
@@ -19,19 +23,34 @@
 #endif
 
 XMALLOCTYPE *
-xmalloc (int n)
+xmalloc (size_t n)
 {
     XMALLOCTYPE *p;
 
     p = (XMALLOCTYPE *) malloc(n);
-    if (p == (XMALLOCTYPE *) 0) {
-       fputs("fetchmail: malloc failed\n",stderr);
-       exit(PS_UNDEFINED);
+    if (p == (XMALLOCTYPE *) 0)
+    {
+       report(stderr, GT_("malloc failed\n"));
+       abort();
     }
     return(p);
 }
 
-char *xstrdup(char *s)
+XMALLOCTYPE *
+xrealloc (XMALLOCTYPE *p, size_t n)
+{
+    if (p == 0)
+       return xmalloc (n);
+    p = (XMALLOCTYPE *) realloc(p, n);
+    if (p == (XMALLOCTYPE *) 0)
+    {
+       report(stderr, GT_("realloc failed\n"));
+       abort();
+    }
+    return p;
+}
+
+char *xstrdup(const char *s)
 {
     char *p;
     p = (char *) xmalloc(strlen(s)+1);
@@ -39,4 +58,27 @@ char *xstrdup(char *s)
     return p;
 }
 
+#if !defined(HAVE_STRDUP)
+char *strdup(const char *s)
+{
+    char *p;
+    p = (char *) malloc(strlen(s)+1);
+    if (p)
+           strcpy(p,s);
+    return p;
+}
+#endif /* !HAVE_STRDUP */
+
+char *xstrndup(const char *s, size_t len)
+{
+    char *p;
+    size_t l = strlen(s);
+
+    if (len < l) l = len;
+    p = (char *)xmalloc(l + 1);
+    memcpy(p, s, l);
+    p[l] = '\0';
+    return p;
+}
+
 /* xmalloc.c ends here */