libstdc++-v3/testsuite/testsuite_hooks.h

   1 // Utility subroutines for the C++ library testsuite.
   2 //
   3 // Copyright (C) 2000, 2001 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 //   #defined before including this header, then no limiting is attempted.
  43
  44 #ifndef _GLIBCPP_TESTSUITE_HOOKS_H
  45 #define _GLIBCPP_TESTSUITE_HOOKS_H
  46
  47 #ifdef DEBUG_ASSERT
  48 # include <cassert>
  49 # define VERIFY(fn) assert(fn)
  50 #else
  51 # define VERIFY(fn) test &= (fn)
  52 # define VERIFY(fn) fn
  53 #endif
  54
  55 #include <bits/c++config.h>
  56
  57 // Defined in GLIBCPP_CONFIGURE_TESTSUITE.
  58 #ifndef _GLIBCPP_MEM_LIMITS
  59
  60 // Don't do memory limits.
  61 void
  62 __set_testsuite_memlimit(float)
  63 { }
  64
  65 #else
  66
  67 // Do memory limits.
  68 #include <sys/resource.h>
  69 #include <unistd.h>
  70
  71 #ifndef MEMLIMIT_MB
  72 #define MEMLIMIT_MB 16.0
  73 #endif
  74
  75 void
  76 __set_testsuite_memlimit(float __size = MEMLIMIT_MB)
  77 {
  78     struct rlimit r;
  79     r.rlim_cur = (rlim_t)(__size * 1048576);
  80
  81     // Heap size, seems to be common.
  82 #if _GLIBCPP_HAVE_MEMLIMIT_DATA
  83     setrlimit(RLIMIT_DATA, &r);
  84 #endif
  85
  86     // Resident set size.
  87 #if _GLIBCPP_HAVE_MEMLIMIT_RSS
  88     setrlimit(RLIMIT_RSS, &r);
  89 #endif
  90
  91     // Mapped memory (brk + mmap).
  92 #if _GLIBCPP_HAVE_MEMLIMIT_VMEM
  93     setrlimit(RLIMIT_VMEM, &r);
  94 #endif
  95
  96     // Virtual memory.
  97 #if _GLIBCPP_HAVE_MEMLIMIT_AS
  98     setrlimit(RLIMIT_AS, &r);
  99 #endif
 100 }
 101 #endif
 102
 103 #endif // _GLIBCPP_TESTSUITE_HOOKS_H
 104