1ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/* 2ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown This file is part of drd, a thread error detector. 3ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 4436e89c602e787e7a27dd6624b09beed41a0da8aDmitriy Ivanov Copyright (C) 2006-2013 Bart Van Assche <bvanassche@acm.org>. 5ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 6ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown This program is free software; you can redistribute it and/or 7ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown modify it under the terms of the GNU General Public License as 8ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown published by the Free Software Foundation; either version 2 of the 9ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown License, or (at your option) any later version. 10ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 11ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown This program is distributed in the hope that it will be useful, but 12ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown WITHOUT ANY WARRANTY; without even the implied warranty of 13ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 14ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown General Public License for more details. 15ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 16ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown You should have received a copy of the GNU General Public License 17ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown along with this program; if not, write to the Free Software 18ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 19ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 02111-1307, USA. 20ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 21ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown The GNU General Public License is contained in the file COPYING. 22ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown*/ 23ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 24ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 25ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include "drd_malloc_wrappers.h" 26ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include "drd_thread.h" 27ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include "pub_tool_basics.h" 28ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include "pub_tool_execontext.h" 29ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include "pub_tool_hashtable.h" 30ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include "pub_tool_libcassert.h" 31ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include "pub_tool_libcbase.h" 32ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include "pub_tool_libcprint.h" 33ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include "pub_tool_mallocfree.h" 34ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include "pub_tool_options.h" 35ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include "pub_tool_replacemalloc.h" 36ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include "pub_tool_threadstate.h" 37ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include "pub_tool_tooliface.h" 38ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 39ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 40ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/* Local type definitions. */ 41ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 42ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/** 43ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown * Node with per-allocation information that will be stored in a hash map. 44ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown * As specified in <pub_tool_hashtable.h>, the first member must be a pointer 45ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown * and the second member must be an UWord. 46ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown */ 47ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Browntypedef struct _DRD_Chunk { 48ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown struct _DRD_Chunk* next; 49ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown UWord data; // pointer to actual block 50ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown SizeT size; // size requested 51ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown ExeContext* where; // where it was allocated 52ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} DRD_Chunk; 53ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 54ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 55ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/* Local variables. */ 56ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 57ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic StartUsingMem s_start_using_mem_callback; 58ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic StopUsingMem s_stop_using_mem_callback; 59ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/* Statistics. */ 60ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic SizeT s_cmalloc_n_mallocs = 0; 61ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic SizeT s_cmalloc_n_frees = 0; 62ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic SizeT s_cmalloc_bs_mallocd = 0; 63ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/* Record malloc'd blocks. */ 64ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic VgHashTable s_malloc_list = NULL; 65ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 66ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 67ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/* Function definitions. */ 68ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 69ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/** Allocate client memory memory and update the hash map. */ 70ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic void* new_block(ThreadId tid, SizeT size, SizeT align, Bool is_zeroed) 71ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 72ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown void* p; 73ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 74ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown p = VG_(cli_malloc)(align, size); 75ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (!p) 76ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return NULL; 77ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (is_zeroed) 78ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown VG_(memset)(p, 0, size); 79ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 80ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown DRD_(malloclike_block)(tid, (Addr)p, size); 81ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 82ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return p; 83ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 84ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 85ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/** 86ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown * Store information about a memory block that has been allocated by 87ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown * malloc() or a malloc() replacement in the hash map. 88ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown */ 89ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownvoid DRD_(malloclike_block)(const ThreadId tid, const Addr p, const SizeT size) 90ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 91ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown DRD_Chunk* mc; 92ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 93ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown tl_assert(p); 94ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 95ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (size > 0) 96ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown s_start_using_mem_callback(p, size, 0/*ec_uniq*/); 97ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 98ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown s_cmalloc_n_mallocs++; 99ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown // Only update this stat if allocation succeeded. 100ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown s_cmalloc_bs_mallocd += size; 101ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 102ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown mc = VG_(malloc)("drd.malloc_wrappers.cDC.1", sizeof(DRD_Chunk)); 103ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown mc->data = p; 104ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown mc->size = size; 105ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown mc->where = VG_(record_ExeContext)(tid, 0); 106ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown VG_(HT_add_node)(s_malloc_list, mc); 107ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 108ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 109ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic void handle_free(ThreadId tid, void* p) 110ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 111b32f58018498ea2225959b0ba11c18f0c433deefEvgeniy Stepanov Bool success; 1129bea4c13fca0e3bb4b719dcb3ed63d47d479294eKenny Root 113b32f58018498ea2225959b0ba11c18f0c433deefEvgeniy Stepanov tl_assert(p); 114b32f58018498ea2225959b0ba11c18f0c433deefEvgeniy Stepanov success = DRD_(freelike_block)(tid, (Addr)p, True); 115b32f58018498ea2225959b0ba11c18f0c433deefEvgeniy Stepanov tl_assert(success); 116ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 117ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 118ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/** 119ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown * Remove the information that was stored by DRD_(malloclike_block)() about 120ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown * a memory block. 121ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown */ 122b32f58018498ea2225959b0ba11c18f0c433deefEvgeniy StepanovBool DRD_(freelike_block)(const ThreadId tid, const Addr p, const Bool dealloc) 123ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 124ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown DRD_Chunk* mc; 125ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 126ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown tl_assert(p); 127ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 128ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown s_cmalloc_n_frees++; 129ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 130b32f58018498ea2225959b0ba11c18f0c433deefEvgeniy Stepanov mc = VG_(HT_lookup)(s_malloc_list, (UWord)p); 131ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (mc) 132ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown { 133ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown tl_assert(p == mc->data); 134ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (mc->size > 0) 135ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown s_stop_using_mem_callback(mc->data, mc->size); 136663860b1408516d02ebfcb3a9999a134e6cfb223Ben Cheng if (dealloc) 137663860b1408516d02ebfcb3a9999a134e6cfb223Ben Cheng VG_(cli_free)((void*)p); 138b32f58018498ea2225959b0ba11c18f0c433deefEvgeniy Stepanov VG_(HT_remove)(s_malloc_list, (UWord)p); 139ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown VG_(free)(mc); 140ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return True; 141ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 142ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return False; 143ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 144ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 145ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/** Wrapper for malloc(). */ 146ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic void* drd_malloc(ThreadId tid, SizeT n) 147ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 148ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return new_block(tid, n, VG_(clo_alignment), /*is_zeroed*/False); 149ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 150ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 151ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/** Wrapper for memalign(). */ 152ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic void* drd_memalign(ThreadId tid, SizeT align, SizeT n) 153ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 154ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return new_block(tid, n, align, /*is_zeroed*/False); 155ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 156ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 157ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/** Wrapper for calloc(). */ 158ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic void* drd_calloc(ThreadId tid, SizeT nmemb, SizeT size1) 159ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 160ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return new_block(tid, nmemb*size1, VG_(clo_alignment), 161ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /*is_zeroed*/True); 162ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 163ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 164ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/** Wrapper for free(). */ 165ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic void drd_free(ThreadId tid, void* p) 166ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 167ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown handle_free(tid, p); 168ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 169ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 170ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/** 171ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown * Wrapper for realloc(). Returns a pointer to the new block of memory, or 172ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown * NULL if no new block could not be allocated. Notes: 173ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown * - realloc(NULL, size) has the same effect as malloc(size). 174ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown * - realloc(p, 0) has the same effect as free(p). 175ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown * - success is not guaranteed even if the requested size is smaller than the 176ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown * allocated size. 177ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown*/ 178ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic void* drd_realloc(ThreadId tid, void* p_old, SizeT new_size) 179ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 180ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown DRD_Chunk* mc; 181ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown void* p_new; 182ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown SizeT old_size; 183ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 184ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (! p_old) 185ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return drd_malloc(tid, new_size); 186ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 187ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (new_size == 0) 188ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown { 189ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown drd_free(tid, p_old); 190ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return NULL; 191ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 192ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 193ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown s_cmalloc_n_mallocs++; 194ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown s_cmalloc_n_frees++; 195ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown s_cmalloc_bs_mallocd += new_size; 196ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 197ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown mc = VG_(HT_lookup)(s_malloc_list, (UWord)p_old); 198ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (mc == NULL) 199ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown { 200ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown tl_assert(0); 201ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return NULL; 202ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 203ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 204ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown old_size = mc->size; 205ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 206ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (old_size == new_size) 207ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown { 208ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* size unchanged */ 209ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown mc->where = VG_(record_ExeContext)(tid, 0); 210ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown p_new = p_old; 211ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 212ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown else if (new_size < old_size) 213ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown { 214ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* new size is smaller but nonzero */ 215ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown s_stop_using_mem_callback(mc->data + new_size, old_size - new_size); 216ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown mc->size = new_size; 217ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown mc->where = VG_(record_ExeContext)(tid, 0); 218ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown p_new = p_old; 219ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 220ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown else 221ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown { 222ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* new size is bigger */ 223ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown p_new = VG_(cli_malloc)(VG_(clo_alignment), new_size); 224ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 225ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (p_new) 226ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown { 227ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* Copy from old to new. */ 228ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown VG_(memcpy)(p_new, p_old, mc->size); 229ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 230ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* Free old memory. */ 231ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (mc->size > 0) 232ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown s_stop_using_mem_callback(mc->data, mc->size); 233663860b1408516d02ebfcb3a9999a134e6cfb223Ben Cheng VG_(cli_free)(p_old); 234ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown VG_(HT_remove)(s_malloc_list, (UWord)p_old); 235ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 236ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* Update state information. */ 237ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown mc->data = (Addr)p_new; 238ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown mc->size = new_size; 239ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown mc->where = VG_(record_ExeContext)(tid, 0); 240ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown VG_(HT_add_node)(s_malloc_list, mc); 241ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown s_start_using_mem_callback((Addr)p_new, new_size, 0/*ec_uniq*/); 242ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 243ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown else 244ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown { 245ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* Allocation failed -- leave original block untouched. */ 246ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 247ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 248ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 249ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return p_new; 250ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 251ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 252ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/** Wrapper for __builtin_new(). */ 253ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic void* drd___builtin_new(ThreadId tid, SizeT n) 254ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 255ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return new_block(tid, n, VG_(clo_alignment), /*is_zeroed*/False); 256ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 257ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 258ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/** Wrapper for __builtin_delete(). */ 259ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic void drd___builtin_delete(ThreadId tid, void* p) 260ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 261ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown handle_free(tid, p); 262ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 263ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 264ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/** Wrapper for __builtin_vec_new(). */ 265ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic void* drd___builtin_vec_new(ThreadId tid, SizeT n) 266ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 267ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return new_block(tid, n, VG_(clo_alignment), /*is_zeroed*/False); 268ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 269ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 270ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/** Wrapper for __builtin_vec_delete(). */ 271ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic void drd___builtin_vec_delete(ThreadId tid, void* p) 272ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 273ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown handle_free(tid, p); 274ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 275ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 276ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/** 277ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown * Wrapper for malloc_usable_size() / malloc_size(). This function takes 278ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown * a pointer to a block allocated by `malloc' and returns the amount of space 279ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown * that is available in the block. This may or may not be more than the size 280ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown * requested from `malloc', due to alignment or minimum size constraints. 281ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown */ 282ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic SizeT drd_malloc_usable_size(ThreadId tid, void* p) 283ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 284ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown DRD_Chunk* mc; 285ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 286ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown mc = VG_(HT_lookup)(s_malloc_list, (UWord)p); 287ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 288ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return mc ? mc->size : 0; 289ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 290ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 291ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownvoid DRD_(register_malloc_wrappers)(const StartUsingMem start_callback, 292ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown const StopUsingMem stop_callback) 293ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 294ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown tl_assert(s_malloc_list == 0); 295ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown s_malloc_list = VG_(HT_construct)("drd_malloc_list"); 296ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown tl_assert(s_malloc_list); 297ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown tl_assert(start_callback); 298ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown tl_assert(stop_callback); 299ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 300ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown s_start_using_mem_callback = start_callback; 301ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown s_stop_using_mem_callback = stop_callback; 302ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 303ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown VG_(needs_malloc_replacement)(drd_malloc, 304ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown drd___builtin_new, 305ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown drd___builtin_vec_new, 306ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown drd_memalign, 307ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown drd_calloc, 308ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown drd_free, 309ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown drd___builtin_delete, 310ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown drd___builtin_vec_delete, 311ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown drd_realloc, 312ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown drd_malloc_usable_size, 313ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 0); 314ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 315ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 316ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff BrownBool DRD_(heap_addrinfo)(Addr const a, 317ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown Addr* const data, 318ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown SizeT* const size, 319ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown ExeContext** const where) 320ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 321ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown DRD_Chunk* mc; 322ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 323ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown tl_assert(data); 324ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown tl_assert(size); 325ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown tl_assert(where); 326ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 327ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown VG_(HT_ResetIter)(s_malloc_list); 328ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown while ((mc = VG_(HT_Next)(s_malloc_list))) 329ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown { 330ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (mc->data <= a && a < mc->data + mc->size) 331ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown { 332ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown *data = mc->data; 333ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown *size = mc->size; 334ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown *where = mc->where; 335ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return True; 336ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 337ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 338ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return False; 339ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 340ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 341ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/*------------------------------------------------------------*/ 342ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/*--- Statistics printing ---*/ 343ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/*------------------------------------------------------------*/ 344ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 345ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownvoid DRD_(print_malloc_stats)(void) 346ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 347ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown DRD_Chunk* mc; 348ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown SizeT nblocks = 0; 349ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown SizeT nbytes = 0; 350ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 351ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (VG_(clo_verbosity) == 0) 352ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return; 353ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (VG_(clo_xml)) 354ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return; 355ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 356ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* Count memory still in use. */ 357ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown VG_(HT_ResetIter)(s_malloc_list); 358ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown while ((mc = VG_(HT_Next)(s_malloc_list))) 359ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown { 360ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown nblocks++; 361ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown nbytes += mc->size; 362ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 363ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 364ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown VG_(message)(Vg_DebugMsg, 365ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown "malloc/free: in use at exit: %lu bytes in %lu blocks.\n", 366ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown nbytes, nblocks); 367ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown VG_(message)(Vg_DebugMsg, 368ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown "malloc/free: %lu allocs, %lu frees, %lu bytes allocated.\n", 369ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown s_cmalloc_n_mallocs, 370ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown s_cmalloc_n_frees, s_cmalloc_bs_mallocd); 371ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (VG_(clo_verbosity) > 1) 372ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown VG_(message)(Vg_DebugMsg, " \n"); 373ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 374ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 375ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/*--------------------------------------------------------------------*/ 376ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/*--- end ---*/ 377ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/*--------------------------------------------------------------------*/ 378