diff mbox

Do we care about malloc failure in tests?

Message ID CALoOobOmtqQ5Dw=6XxGO+K9OFEvcwq8kR0BgJnJpsxxKN7zHQg@mail.gmail.com
State Committed
Headers

Commit Message

Paul Pluzhnikov June 7, 2016, 3:51 p.m. UTC 
  On Sun, Jun 5, 2016 at 7:12 PM, Mike Frysinger <vapier@gentoo.org> wrote:
> On 05 Jun 2016 19:00, Paul Pluzhnikov wrote:
>> I think I can also remove this from xrealloc, since our realloc does
>> the right thing when p==NULL.
>>
>> +  if (p == 0)
>> +    return xmalloc (n);
>
> right, we don't want/need that
>
> i think all the use of 0 should be changed to NULL too

I'll submit attached patch tomorrow if there are no further comments.

Thanks,

2016-06-07  Paul Pluzhnikov  <ppluzhnikov@google.com>

        * test-skeleton.c (oom_error, xmalloc, xcalloc, xrealloc):
        New functions.
        (add_temp_file): Use them.
diff mbox

Patch

diff --git a/test-skeleton.c b/test-skeleton.c
index 29bdc9c..e091d7f 100644
--- a/test-skeleton.c
+++ b/test-skeleton.c
@@ -70,6 +70,51 @@  static pid_t pid;
 /* Directory to place temporary files in.  */
 static const char *test_dir;
 
+static void
+oom_error (const char *fn)
+{
+  printf ("%s: allocation failed: %m\n", fn);
+  exit (1);
+}
+
+/* Allocate N bytes of memory dynamically, with error checking.  */
+static void *
+__attribute__ ((used))
+xmalloc (size_t n)
+{
+  void *p;
+
+  p = malloc (n);
+  if (p == NULL)
+    oom_error ("malloc");
+  return p;
+}
+
+/* Allocate memory for N elements of S bytes, with error checking.  */
+static void *
+__attribute__ ((used))
+xcalloc (size_t n, size_t s)
+{
+  void *p;
+
+  p = calloc (n, s);
+  if (p == NULL)
+    oom_error ("calloc");
+  return p;
+}
+
+/* Change the size of an allocated block of memory P to N bytes,
+   with error checking.  */
+static void *
+__attribute__ ((used))
+xrealloc (void *p, size_t n)
+{
+  p = realloc (p, n);
+  if (p == NULL)
+    oom_error ("realloc");
+  return p;
+}
+
 /* List of temporary files.  */
 struct temp_name_list
 {
@@ -83,9 +128,9 @@  __attribute__ ((unused))
 add_temp_file (const char *name)
 {
   struct temp_name_list *newp
-    = (struct temp_name_list *) calloc (sizeof (*newp), 1);
+    = (struct temp_name_list *) xcalloc (sizeof (*newp), 1);
   char *newname = strdup (name);
-  if (newp != NULL && newname != NULL)
+  if (newname != NULL)
     {
       newp->name = newname;
       if (temp_name_list == NULL)
@@ -124,13 +169,8 @@  create_temp_file (const char *base, char **filename)
   char *fname;
   int fd;
 
-  fname = (char *) malloc (strlen (test_dir) + 1 + strlen (base)
-			   + sizeof ("XXXXXX"));
-  if (fname == NULL)
-    {
-      puts ("out of memory");
-      return -1;
-    }
+  fname = (char *) xmalloc (strlen (test_dir) + 1 + strlen (base)
+			    + sizeof ("XXXXXX"));
   strcpy (stpcpy (stpcpy (stpcpy (fname, test_dir), "/"), base), "XXXXXX");
 
   fd = mkstemp (fname);