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