david/ipxe
david/ipxe
Archived
1
0
Fork 0
Code

[malloc] Report caller address as soon as memory corruption is detected

Browse Source 
Signed-off-by: Michael Brown <mcb30@ipxe.org>
This commit is contained in:
Michael Brown 2014-12-15 14:48:02 +00:00
parent 7871666740
commit 35c5379760
1 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/core/malloc.c
View File

@ -517,6 +517,10 @@ void * realloc ( void *old_ptr, size_t new_size ) {
VALGRIND_FREELIKE_BLOCK ( old_ptr, 0 ); VALGRIND_FREELIKE_BLOCK ( old_ptr, 0 );
} }
if ( ASSERTED ) {
DBGC ( &heap, "Possible memory corruption detected from %p\n",
__builtin_return_address ( 0 ) );
}
return new_ptr; return new_ptr;
} }
@ -530,7 +534,14 @@ void * realloc ( void *old_ptr, size_t new_size ) {
* will be aligned to at least a multiple of sizeof(void*). * will be aligned to at least a multiple of sizeof(void*).
*/ */
void * malloc ( size_t size ) { void * malloc ( size_t size ) {
return realloc ( NULL, size ); void *ptr;
ptr = realloc ( NULL, size );
if ( ASSERTED ) {
DBGC ( &heap, "Possible memory corruption detected from %p\n",
__builtin_return_address ( 0 ) );
}
return ptr;
} }
/** /**
@ -544,7 +555,12 @@ void * malloc ( size_t size ) {
* If @c ptr is NULL, no action is taken. * If @c ptr is NULL, no action is taken.
*/ */
void free ( void *ptr ) { void free ( void *ptr ) {
realloc ( ptr, 0 ); realloc ( ptr, 0 );
if ( ASSERTED ) {
DBGC ( &heap, "Possible memory corruption detected from %p\n",
__builtin_return_address ( 0 ) );
}
} }
/** /**
@ -564,6 +580,10 @@ void * zalloc ( size_t size ) {
data = malloc ( size ); data = malloc ( size );
if ( data ) if ( data )
memset ( data, 0, size ); memset ( data, 0, size );
if ( ASSERTED ) {
DBGC ( &heap, "Possible memory corruption detected from %p\n",
__builtin_return_address ( 0 ) );
}
return data; return data;
} }