libstdc++-v3/testsuite/ext/profile/mh.cc

   1 // { dg-do compile { target *-*-linux* *-*-gnu* } }
   2 // { dg-xfail-if "" { uclibc } { "*" } { "" } }
   3 // { dg-require-profile-mode "" }
   4
   5 // -*- C++ -*-
   6
   7 // Copyright (C) 2006-2014 Free Software Foundation, Inc.
   8 //
   9 // This file is part of the GNU ISO C++ Library.  This library is free
  10 // software; you can redistribute it and/or modify it under the
  11 // terms of the GNU General Public License as published by the
  12 // Free Software Foundation; either version 3, or (at your option)
  13 // any later version.
  14 //
  15 // This library is distributed in the hope that it will be useful,
  16 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  17 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18 // GNU General Public License for more details.
  19 //
  20 // You should have received a copy of the GNU General Public License along
  21 // with this library; see the file COPYING3.  If not see
  22 // <http://www.gnu.org/licenses/>.
  23
  24 #include <stdio.h>
  25 #include <malloc.h>
  26 #include <vector>
  27
  28 using std::vector;
  29
  30 static void my_init_hook (void);
  31 static void *my_malloc_hook (size_t, const void *);
  32 typedef void* (*malloc_hook) (size_t, const void *);
  33
  34 malloc_hook old_malloc_hook;
  35
  36 void (*__malloc_initialize_hook) (void) = my_init_hook;
  37
  38 static void
  39 my_init_hook (void)
  40 {
  41   old_malloc_hook = __malloc_hook;
  42   __malloc_hook = my_malloc_hook;
  43 }
  44
  45 static void *
  46 my_malloc_hook (size_t size, const void *caller)
  47 {
  48   void *result;
  49   __malloc_hook = old_malloc_hook;
  50   result = malloc (size);
  51   old_malloc_hook = __malloc_hook;
  52
  53   // With _GLIBCXX_PROFILE, the instrumentation of the vector constructor
  54   // will call back into malloc.
  55   vector<int> v;
  56
  57   __malloc_hook = my_malloc_hook;
  58   return result;
  59 }
  60
  61
  62 int main()
  63 {
  64   int* test = (int*) malloc(sizeof(int));
  65   *test = 1;
  66   return *test;
  67 }