libstdc++-v3/testsuite/testsuite_hooks.h

   1 // Utility subroutines for the C++ library testsuite.
   2 //
   3 // Copyright (C) 2000, 2001, 2002 Free Software Foundation, Inc.
   4 //
   5 // This file is part of the GNU ISO C++ Library.  This library is free
   6 // software; you can redistribute it and/or modify it under the
   7 // terms of the GNU General Public License as published by the
   8 // Free Software Foundation; either version 2, or (at your option)
   9 // 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
  14 // GNU General Public License for more details.
  15 //
  16 // You should have received a copy of the GNU General Public License along
  17 // with this library; see the file COPYING.  If not, write to the Free
  18 // Software Foundation, 59 Temple Place - Suite 330, Boston, MA 02111-1307,
  19 // USA.
  20 //
  21 // As a special exception, you may use this file as part of a free software
  22 // library without restriction.  Specifically, if other files instantiate
  23 // templates or use macros or inline functions from this file, or you compile
  24 // this file and link it with other files to produce an executable, this
  25 // file does not by itself cause the resulting executable to be covered by
  26 // the GNU General Public License.  This exception does not however
  27 // invalidate any other reasons why the executable file might be covered by
  28 // the GNU General Public License.
  29
  30 // This file provides the following:
  31 //
  32 // 1)  VERIFY(), via DEBUG_ASSERT, from Brent Verner <brent@rcfile.org>.
  33 //   This file is included in the various testsuite programs to provide
  34 //   #define(able) assert() behavior for debugging/testing. It may be
  35 //   a suitable location for other furry woodland creatures as well.
  36 //
  37 // 2)  __set_testsuite_memlimit()
  38 //   __set_testsuite_memlimit() uses setrlimit() to restrict dynamic memory
  39 //   allocation.  We provide a default memory limit if none is passed by the
  40 //   calling application.  The argument to __set_testsuite_memlimit() is the
  41 //   limit in megabytes (a floating-point number).  If _GLIBCPP_MEM_LIMITS is
  42 //   not #defined before including this header, then no limiting is attempted.
  43 //
  44 // 3)  gnu_counting_struct
  45 //   This is a POD with a static data member, gnu_counting_struct::count,
  46 //   which starts at zero, increments on instance construction, and decrements
  47 //   on instance destruction.  "assert_count(n)" can be called to VERIFY()
  48 //   that the count equals N.
  49
  50 #ifndef _GLIBCPP_TESTSUITE_HOOKS_H
  51 #define _GLIBCPP_TESTSUITE_HOOKS_H
  52
  53 #ifdef DEBUG_ASSERT
  54 # include <cassert>
  55 # define VERIFY(fn) assert(fn)
  56 #else
  57 # define VERIFY(fn) test &= (fn)
  58 #endif
  59
  60 #include <bits/c++config.h>
  61
  62 // Defined in GLIBCPP_CONFIGURE_TESTSUITE.
  63 #ifndef _GLIBCPP_MEM_LIMITS
  64
  65 // Don't do memory limits.
  66 void
  67 __set_testsuite_memlimit(float x = 0)
  68 { }
  69
  70 #else
  71
  72 // Do memory limits.
  73 #include <sys/resource.h>
  74 #include <unistd.h>
  75
  76 #ifndef MEMLIMIT_MB
  77 #define MEMLIMIT_MB 16.0
  78 #endif
  79
  80 void
  81 __set_testsuite_memlimit(float __size = MEMLIMIT_MB)
  82 {
  83     struct rlimit r;
  84     rlim_t limit = (rlim_t)(__size * 1048576);
  85
  86     // Heap size, seems to be common.
  87 #if _GLIBCPP_HAVE_MEMLIMIT_DATA
  88     getrlimit(RLIMIT_DATA, &r);
  89     r.rlim_cur = limit;
  90     setrlimit(RLIMIT_DATA, &r);
  91 #endif
  92
  93     // Resident set size.
  94 #if _GLIBCPP_HAVE_MEMLIMIT_RSS
  95     getrlimit(RLIMIT_RSS, &r);
  96     r.rlim_cur = limit;
  97     setrlimit(RLIMIT_RSS, &r);
  98 #endif
  99
 100     // Mapped memory (brk + mmap).
 101 #if _GLIBCPP_HAVE_MEMLIMIT_VMEM
 102     getrlimit(RLIMIT_VMEM, &r);
 103     r.rlim_cur = limit;
 104     setrlimit(RLIMIT_VMEM, &r);
 105 #endif
 106
 107     // Virtual memory.
 108 #if _GLIBCPP_HAVE_MEMLIMIT_AS
 109     getrlimit(RLIMIT_AS, &r);
 110     r.rlim_cur = limit;
 111     setrlimit(RLIMIT_AS, &r);
 112 #endif
 113 }
 114 #endif
 115
 116
 117 struct gnu_counting_struct
 118 {
 119     // Specifically and glaringly-obviously marked 'signed' so that when
 120     // count mistakenly goes negative, we can track the patterns of
 121     // deletions easier.
 122     typedef  signed int     size_type;
 123     static size_type   count;
 124     gnu_counting_struct() { ++count; }
 125     gnu_counting_struct (const gnu_counting_struct&) { ++count; }
 126     ~gnu_counting_struct() { --count; }
 127 };
 128
 129 #define assert_count(n)   VERIFY(gnu_counting_struct::count == n)
 130
 131 gnu_counting_struct::size_type  gnu_counting_struct::count = 0;
 132
 133
 134 #endif // _GLIBCPP_TESTSUITE_HOOKS_H
 135