tests/test-safe-alloc.c

   1 /*
   2  * Test the safe-alloc macros
   3  *
   4  * Copyright (C) 2009 Free Software Foundation, Inc.
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307  USA
  19  *
  20  * Author: David Lutterkort <lutter@redhat.com>
  21  */
  22
  23 #include <config.h>
  24
  25 #include "safe-alloc.h"
  26
  27 #include <stdio.h>
  28 #include <stdlib.h>
  29
  30 #define ASSERT(expr) \
  31   do                                                                         \
  32     {                                                                        \
  33       if (!(expr))                                                           \
  34         {                                                                    \
  35           fprintf (stderr, "%s:%d: assertion failed\n", __FILE__, __LINE__); \
  36           fflush (stderr);                                                   \
  37           abort ();                                                          \
  38         }                                                                    \
  39     }                                                                        \
  40   while (0)
  41
  42 int
  43 main ()
  44 {
  45   struct tst
  46   {
  47     int a;
  48     int b;
  49   };
  50
  51   struct tst *p = NULL;
  52   int r;
  53
  54   r = ALLOC (p);
  55   ASSERT (r >= 0);
  56
  57   ASSERT (p->a == 0 && p->b == 0);
  58
  59   p->a = p->b = 42;
  60   r = REALLOC_N (p, 5);
  61
  62   ASSERT (p[0].a == 42 && p[0].b == 42);
  63
  64   FREE (p);
  65   ASSERT (p == NULL);
  66
  67   return 0;
  68 }