Home | History | Annotate | Download | only in m_replacemalloc
      1 
      2 /*--------------------------------------------------------------------*/
      3 /*--- Malloc replacement.                     replacemalloc_core.c ---*/
      4 /*--------------------------------------------------------------------*/
      5 
      6 /*
      7    This file is part of Valgrind, a dynamic binary instrumentation
      8    framework.
      9 
     10    Copyright (C) 2000-2012 Julian Seward
     11       jseward (at) acm.org
     12 
     13    This program is free software; you can redistribute it and/or
     14    modify it under the terms of the GNU General Public License as
     15    published by the Free Software Foundation; either version 2 of the
     16    License, or (at your option) any later version.
     17 
     18    This program is distributed in the hope that it will be useful, but
     19    WITHOUT ANY WARRANTY; without even the implied warranty of
     20    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
     21    General Public License for more details.
     22 
     23    You should have received a copy of the GNU General Public License
     24    along with this program; if not, write to the Free Software
     25    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
     26    02111-1307, USA.
     27 
     28    The GNU General Public License is contained in the file COPYING.
     29 */
     30 
     31 #include "pub_core_basics.h"
     32 #include "pub_core_libcbase.h"
     33 #include "pub_core_libcprint.h"
     34 #include "pub_core_mallocfree.h"
     35 #include "pub_core_options.h"
     36 #include "pub_core_replacemalloc.h"
     37 
     38 /*------------------------------------------------------------*/
     39 /*--- Command line options                                 ---*/
     40 /*------------------------------------------------------------*/
     41 
     42 /* Nb: the allocator always rounds blocks up to a multiple of
     43    VG_MIN_MALLOC_SZB.
     44 */
     45 
     46 /* DEBUG: print malloc details?  default: NO */
     47 Bool VG_(clo_trace_malloc)  = False;
     48 
     49 /* Minimum alignment in functions that don't specify alignment explicitly.
     50    default: 0, i.e. use VG_MIN_MALLOC_SZB. */
     51 UInt VG_(clo_alignment)     = VG_MIN_MALLOC_SZB;
     52 
     53 
     54 Bool VG_(replacement_malloc_process_cmd_line_option)(Char* arg)
     55 {
     56    if VG_INT_CLO(arg, "--alignment", VG_(clo_alignment)) {
     57       if (VG_(clo_alignment) < VG_MIN_MALLOC_SZB ||
     58           VG_(clo_alignment) > 4096 ||
     59           VG_(log2)( VG_(clo_alignment) ) == -1 /* not a power of 2 */)
     60       {
     61          VG_(fmsg_bad_option)(arg,
     62             "Alignment must be a power of 2 in the range %d..4096.\n",
     63             VG_MIN_MALLOC_SZB);
     64       }
     65    }
     66 
     67    else if VG_BOOL_CLO(arg, "--trace-malloc",  VG_(clo_trace_malloc)) {}
     68    else
     69       return False;
     70 
     71    return True;
     72 }
     73 
     74 /*------------------------------------------------------------*/
     75 /*--- Useful functions                                     ---*/
     76 /*------------------------------------------------------------*/
     77 
     78 void* VG_(cli_malloc) ( SizeT align, SizeT nbytes )
     79 {
     80    // 'align' should be valid (ie. big enough and a power of two) by now.
     81    // VG_(arena_memalign)() will abort if it's not.
     82    if (VG_MIN_MALLOC_SZB == align)
     83       return VG_(arena_malloc)   ( VG_AR_CLIENT, "replacemalloc.cm.1",
     84                                    nbytes );
     85    else
     86       return VG_(arena_memalign) ( VG_AR_CLIENT, "replacemalloc.cm.2",
     87                                    align, nbytes );
     88 }
     89 
     90 void VG_(cli_free) ( void* p )
     91 {
     92    VG_(arena_free) ( VG_AR_CLIENT, p );
     93 }
     94 
     95 Bool VG_(addr_is_in_block)( Addr a, Addr start, SizeT size, SizeT rz_szB )
     96 {
     97    return ( start - rz_szB <= a  &&  a < start + size + rz_szB );
     98 }
     99 
    100 /*--------------------------------------------------------------------*/
    101 /*--- end                                                          ---*/
    102 /*--------------------------------------------------------------------*/
    103