pintos-os.org Git - pspp/blob - 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 <stdio.h>
  26
  27 #include <safe-alloc.h>
  28
  29 #define ASSERT(expr) \
  30   do                                                                         \
  31     {                                                                        \
  32       if (!(expr))                                                           \
  33         {                                                                    \
  34           fprintf (stderr, "%s:%d: assertion failed\n", __FILE__, __LINE__); \
  35           fflush (stderr);                                                   \
  36           abort ();                                                          \
  37         }                                                                    \
  38     }                                                                        \
  39   while (0)
  40
  41 int main()
  42 {
  43   struct tst {
  44     int a;
  45     int b;
  46   };
  47
  48   struct tst *p = NULL;
  49   int r;
  50
  51   r = ALLOC(p);
  52   ASSERT(r >= 0);
  53
  54   ASSERT(p->a == 0 && p->b == 0);
  55
  56   p->a = p->b = 42;
  57   r = REALLOC_N(p, 5);
  58
  59   ASSERT(p[0].a == 42 && p[0].b == 42);
  60
  61   FREE(p);
  62   ASSERT(p == NULL);
  63 }