malloc_debug_qemu.cpp revision 0d787c1fa18c6a1f29ef9840e28a68cf077be1de
1/* 2 * Copyright (C) 2009 The Android Open Source Project 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * * Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * * Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in 12 * the documentation and/or other materials provided with the 13 * distribution. 14 * 15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 16 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 17 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS 18 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE 19 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, 20 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, 21 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS 22 * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED 23 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, 24 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT 25 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 26 * SUCH DAMAGE. 27 */ 28 29/* 30 * Contains implementation of memory allocation routines instrumented for 31 * usage in the emulator to detect memory allocation violations, such as 32 * memory leaks, buffer overruns, etc. 33 * Code, implemented here is intended to run in the emulated environment only, 34 * and serves simply as hooks into memory allocation routines. Main job of this 35 * code is to notify the emulator about memory being allocated/deallocated, 36 * providing information about each allocation. The idea is that emulator will 37 * keep list of currently allocated blocks, and, knowing boundaries of each 38 * block it will be able to verify that ld/st access to these blocks don't step 39 * over boundaries set for the user. To enforce that, each memory block 40 * allocated by this code is guarded with "prefix" and "suffix" areas, so 41 * every time emulator detects access to any of these guarding areas, it can be 42 * considered as access violation. 43 */ 44 45#include <stdlib.h> 46#include <stddef.h> 47#include <stdio.h> 48#include <fcntl.h> 49#include <sys/mman.h> 50#include <pthread.h> 51#include <unistd.h> 52#include <errno.h> 53#include "dlmalloc.h" 54#include "libc_logging.h" 55#include "malloc_debug_common.h" 56 57/* This file should be included into the build only when 58 * MALLOC_QEMU_INSTRUMENT macro is defined. */ 59#ifndef MALLOC_QEMU_INSTRUMENT 60#error MALLOC_QEMU_INSTRUMENT is not defined. 61#endif // !MALLOC_QEMU_INSTRUMENT 62 63/* Controls access violation test performed to make sure that we catch AVs 64 * all the time they occur. See test_access_violation for more info. This macro 65 * is used for internal testing purposes and should always be set to zero for 66 * the production builds. */ 67#define TEST_ACCESS_VIOLATIONS 0 68 69// ============================================================================= 70// Communication structures 71// ============================================================================= 72 73/* Describes memory block allocated from the heap. This structure is passed 74 * along with TRACE_DEV_REG_MALLOC event. This descriptor is used to inform 75 * the emulator about new memory block being allocated from the heap. The entire 76 * structure is initialized by the guest system before event is fired up. It is 77 * important to remember that same structure (an exact copy, except for 78 * replacing pointers with target_ulong) is also declared in the emulator's 79 * sources (file memcheck/memcheck_common.h). So, every time a change is made to 80 * any of these two declaration, another one must be also updated accordingly. 81 */ 82struct MallocDesc { 83 /* Pointer to the memory block actually allocated from the heap. Note that 84 * this is not the pointer that is returned to the malloc's caller. Pointer 85 * returned to the caller is calculated by adding value stored in this field 86 * to the value stored in prefix_size field of this structure. 87 */ 88 void* ptr; 89 90 /* Number of bytes requested by the malloc's caller. */ 91 uint32_t requested_bytes; 92 93 /* Byte size of the prefix data. Actual pointer returned to the malloc's 94 * caller is calculated by adding value stored in this field to the value 95 * stored in in the ptr field of this structure. 96 */ 97 uint32_t prefix_size; 98 99 /* Byte size of the suffix data. */ 100 uint32_t suffix_size; 101 102 /* Id of the process that initialized libc instance, in which allocation 103 * has occurred. This field is used by the emulator to report errors in 104 * the course of TRACE_DEV_REG_MALLOC event handling. In case of an error, 105 * emulator sets this field to zero (invalid value for a process ID). 106 */ 107 uint32_t libc_pid; 108 109 /* Id of the process in context of which allocation has occurred. 110 * Value in this field may differ from libc_pid value, if process that 111 * is doing allocation has been forked from the process that initialized 112 * libc instance. 113 */ 114 uint32_t allocator_pid; 115 116 /* Number of access violations detected on this allocation. */ 117 uint32_t av_count; 118}; 119 120/* Describes memory block info queried from emulator. This structure is passed 121 * along with TRACE_DEV_REG_QUERY_MALLOC event. When handling free and realloc 122 * calls, it is required that we have information about memory blocks that were 123 * actually allocated in previous calls to malloc, calloc, memalign, or realloc. 124 * Since we don't keep this information directly in the allocated block, but 125 * rather we keep it in the emulator, we need to query emulator for that 126 * information with TRACE_DEV_REG_QUERY_MALLOC query. The entire structure is 127 * initialized by the guest system before event is fired up. It is important to 128 * remember that same structure (an exact copy, except for replacing pointers 129 * with target_ulong) is also declared in the emulator's sources (file 130 * memcheck/memecheck_common.h). So, every time a change is made to any of these 131 * two declaration, another one must be also updated accordingly. 132 */ 133struct MallocDescQuery { 134 /* Pointer, for which information is queried. Note that this pointer doesn't 135 * have to be exact pointer returned to malloc's caller, but can point 136 * anywhere inside an allocated block, including guarding areas. Emulator 137 * will respond with information about allocated block that contains this 138 * pointer. 139 */ 140 void* ptr; 141 142 /* Id of the process that initialized libc instance, in which this query 143 * is called. This field is used by the emulator to report errors in 144 * the course of TRACE_DEV_REG_QUERY_MALLOC event handling. In case of an 145 * error, emulator sets this field to zero (invalid value for a process ID). 146 */ 147 uint32_t libc_pid; 148 149 /* Process ID in context of which query is made. */ 150 uint32_t query_pid; 151 152 /* Code of the allocation routine, in context of which query has been made: 153 * 1 - free 154 * 2 - realloc 155 */ 156 uint32_t routine; 157 158 /* Address of memory allocation descriptor for the queried pointer. 159 * Descriptor, addressed by this field is initialized by the emulator in 160 * response to the query. 161 */ 162 MallocDesc* desc; 163}; 164 165/* Describes memory block that is being freed back to the heap. This structure 166 * is passed along with TRACE_DEV_REG_FREE_PTR event. The entire structure is 167 * initialized by the guest system before event is fired up. It is important to 168 * remember that same structure (an exact copy, except for replacing pointers 169 * with target_ulong) is also declared in the emulator's sources (file 170 * memcheck/memecheck_common.h). So, every time a change is made to any of these 171 * two declaration, another one must be also updated accordingly. 172 */ 173struct MallocFree { 174 /* Pointer to be freed. */ 175 void* ptr; 176 177 /* Id of the process that initialized libc instance, in which this free 178 * is called. This field is used by the emulator to report errors in 179 * the course of TRACE_DEV_REG_FREE_PTR event handling. In case of an 180 * error, emulator sets this field to zero (invalid value for a process ID). 181 */ 182 uint32_t libc_pid; 183 184 /* Process ID in context of which memory is being freed. */ 185 uint32_t free_pid; 186}; 187 188// ============================================================================= 189// Communication events 190// ============================================================================= 191 192/* Notifies the emulator that libc has been initialized for a process. 193 * Event's value parameter is PID for the process in context of which libc has 194 * been initialized. 195 */ 196#define TRACE_DEV_REG_LIBC_INIT 1536 197 198/* Notifies the emulator about new memory block been allocated. 199 * Event's value parameter points to MallocDesc instance that contains 200 * allocated block information. Note that 'libc_pid' field of the descriptor 201 * is used by emulator to report failure in handling this event. In case 202 * of a failure emulator will zero that field before completing this event. 203 */ 204#define TRACE_DEV_REG_MALLOC 1537 205 206/* Notifies the emulator about memory block being freed. 207 * Event's value parameter points to MallocFree descriptor that contains 208 * information about block that's being freed. Note that 'libc_pid' field 209 * of the descriptor is used by emulator to report failure in handling this 210 * event. In case of a failure emulator will zero that field before completing 211 * this event. 212 */ 213#define TRACE_DEV_REG_FREE_PTR 1538 214 215/* Queries the emulator about allocated memory block information. 216 * Event's value parameter points to MallocDescQuery descriptor that contains 217 * query parameters. Note that 'libc_pid' field of the descriptor is used by 218 * emulator to report failure in handling this event. In case of a failure 219 * emulator will zero that field before completing this event. 220 */ 221#define TRACE_DEV_REG_QUERY_MALLOC 1539 222 223/* Queries the emulator to print a string to its stdout. 224 * Event's value parameter points to a zero-terminated string to be printed. 225 */ 226#define TRACE_DEV_REG_PRINT_USER_STR 1540 227 228static void notify_qemu_string(const char* str); 229static void qemu_log(int prio, const char* fmt, ...); 230static void dump_malloc_descriptor(char* str, 231 size_t str_buf_size, 232 const MallocDesc* desc); 233 234// ============================================================================= 235// Macros 236// ============================================================================= 237 238/* Defines default size of allocation prefix. 239 * Note that we make prefix area quite large in order to increase chances of 240 * catching buffer overflow. */ 241#define DEFAULT_PREFIX_SIZE (malloc_alignment * 4) 242 243/* Defines default size of allocation suffix. 244 * Note that we make suffix area quite large in order to increase chances of 245 * catching buffer overflow. */ 246#define DEFAULT_SUFFIX_SIZE (malloc_alignment * 4) 247 248/* Debug tracing has been enabled by the emulator. */ 249#define DEBUG_TRACING_ENABLED 0x00000001 250/* Error tracing has been enabled by the emulator. */ 251#define ERROR_TRACING_ENABLED 0x00000002 252/* Info tracing has been enabled by the emulator. */ 253#define INFO_TRACING_ENABLED 0x00000004 254/* All tracing flags combined. */ 255#define ALL_TRACING_ENABLED (DEBUG_TRACING_ENABLED | \ 256 ERROR_TRACING_ENABLED | \ 257 INFO_TRACING_ENABLED) 258 259/* Prints a string to the emulator's stdout. 260 * In early stages of system loading, logging messages to logcat 261 * is not available, because ADB API has not been 262 * hooked up yet. So, in order to see such messages we need to print them to 263 * the emulator's stdout. 264 * Parameters passed to this macro are the same as parameters for printf 265 * routine. 266 */ 267#define TR(...) \ 268 do { \ 269 char tr_str[4096]; \ 270 snprintf(tr_str, sizeof(tr_str), __VA_ARGS__); \ 271 tr_str[sizeof(tr_str) - 1] = '\0'; \ 272 notify_qemu_string(&tr_str[0]); \ 273 } while (0) 274 275// ============================================================================= 276// Logging macros. Note that we simultaneously log messages to ADB and emulator. 277// ============================================================================= 278 279/* 280 * Helper macros for checking if particular trace level is enabled. 281 */ 282#define debug_LOG_ENABLED ((tracing_flags & DEBUG_TRACING_ENABLED) != 0) 283#define error_LOG_ENABLED ((tracing_flags & ERROR_TRACING_ENABLED) != 0) 284#define info_LOG_ENABLED ((tracing_flags & INFO_TRACING_ENABLED) != 0) 285#define tracing_enabled(type) (type##_LOG_ENABLED) 286 287/* 288 * Logging helper macros. 289 */ 290#define qemu_debug_log(format, ...) \ 291 do { \ 292 __libc_format_log(ANDROID_LOG_DEBUG, "memcheck", (format), ##__VA_ARGS__); \ 293 if (tracing_flags & DEBUG_TRACING_ENABLED) { \ 294 qemu_log(ANDROID_LOG_DEBUG, (format), ##__VA_ARGS__); \ 295 } \ 296 } while (0) 297 298#define qemu_error_log(format, ...) \ 299 do { \ 300 __libc_format_log(ANDROID_LOG_ERROR, "memcheck", (format), ##__VA_ARGS__); \ 301 if (tracing_flags & ERROR_TRACING_ENABLED) { \ 302 qemu_log(ANDROID_LOG_ERROR, (format), ##__VA_ARGS__); \ 303 } \ 304 } while (0) 305 306#define qemu_info_log(format, ...) \ 307 do { \ 308 __libc_format_log(ANDROID_LOG_INFO, "memcheck", (format), ##__VA_ARGS__); \ 309 if (tracing_flags & INFO_TRACING_ENABLED) { \ 310 qemu_log(ANDROID_LOG_INFO, (format), ##__VA_ARGS__); \ 311 } \ 312 } while (0) 313 314/* Logs message dumping MallocDesc instance at the end of the message. 315 * Param: 316 * type - Message type: debug, error, or info 317 * desc - MallocDesc instance to dump. 318 * fmt + rest - Formats message preceding dumped descriptor. 319*/ 320#define log_mdesc(type, desc, fmt, ...) \ 321 do { \ 322 if (tracing_enabled(type)) { \ 323 char log_str[4096]; \ 324 __libc_format_buffer(log_str, sizeof(log_str), fmt, ##__VA_ARGS__); \ 325 log_str[sizeof(log_str) - 1] = '\0'; \ 326 size_t str_len = strlen(log_str); \ 327 dump_malloc_descriptor(log_str + str_len, \ 328 sizeof(log_str) - str_len, \ 329 (desc)); \ 330 type##_log("%s", log_str); \ 331 } \ 332 } while (0) 333 334// ============================================================================= 335// Static data 336// ============================================================================= 337 338/* Emulator's magic page address. 339 * This page (mapped on /dev/qemu_trace device) is used to fire up events 340 * in the emulator. */ 341static volatile void* qtrace = NULL; 342 343/* Cached PID of the process in context of which this libc instance 344 * has been initialized. */ 345static uint32_t malloc_pid = 0; 346 347/* Memory allocation alignment that is used in dlmalloc. 348 * This variable is updated by memcheck_initialize routine. */ 349static uint32_t malloc_alignment = 8; 350 351/* Tracing flags. These flags control which types of logging messages are 352 * enabled by the emulator. See XXX_TRACING_ENABLED for the values of flags 353 * stored in this variable. This variable is updated by memcheck_initialize 354 * routine. */ 355static uint32_t tracing_flags = 0; 356 357// ============================================================================= 358// Static routines 359// ============================================================================= 360 361/* Gets pointer, returned to malloc caller for the given allocation decriptor. 362 * Param: 363 * desc - Allocation descriptor. 364 * Return: 365 * Pointer to the allocated memory returned to the malloc caller. 366 */ 367static inline void* mallocdesc_user_ptr(const MallocDesc* desc) { 368 return static_cast<char*>(desc->ptr) + desc->prefix_size; 369} 370 371/* Gets size of memory block actually allocated from the heap for the given 372 * allocation decriptor. 373 * Param: 374 * desc - Allocation descriptor. 375 * Return: 376 * Size of memory block actually allocated from the heap. 377 */ 378static inline uint32_t mallocdesc_alloc_size(const MallocDesc* desc) { 379 return desc->prefix_size + desc->requested_bytes + desc->suffix_size; 380} 381 382/* Gets pointer to the end of the allocated block for the given descriptor. 383 * Param: 384 * desc - Descriptor for the memory block, allocated in malloc handler. 385 * Return: 386 * Pointer to the end of (one byte past) the allocated block. 387 */ 388static inline void* mallocdesc_alloc_end(const MallocDesc* desc) { 389 return static_cast<char*>(desc->ptr) + mallocdesc_alloc_size(desc); 390} 391 392/* Fires up an event in the emulator. 393 * Param: 394 * code - Event code (one of the TRACE_DEV_XXX). 395 * val - Event's value parameter. 396 */ 397static inline void notify_qemu(uint32_t code, uint32_t val) { 398 if (NULL != qtrace) { 399 *(volatile uint32_t*)((uint32_t)qtrace + ((code - 1024) << 2)) = val; 400 } 401} 402 403/* Prints a zero-terminated string to the emulator's stdout (fires up 404 * TRACE_DEV_REG_PRINT_USER_STR event in the emulator). 405 * Param: 406 * str - Zero-terminated string to print. 407 */ 408static void notify_qemu_string(const char* str) { 409 if (str != NULL) { 410 notify_qemu(TRACE_DEV_REG_PRINT_USER_STR, (uint32_t)str); 411 } 412} 413 414/* Fires up TRACE_DEV_REG_LIBC_INIT event in the emulator. 415 * Param: 416 * pid - ID of the process that initialized libc. 417 */ 418static void notify_qemu_libc_initialized(uint32_t pid) { 419 notify_qemu(TRACE_DEV_REG_LIBC_INIT, pid); 420} 421 422/* Fires up TRACE_DEV_REG_MALLOC event in the emulator. 423 * Param: 424 * desc - Pointer to MallocDesc instance containing allocated block 425 * information. 426 * Return: 427 * Zero on success, or -1 on failure. Note that on failure libc_pid field of 428 * the desc parameter passed to this routine has been zeroed out by the 429 * emulator. 430 */ 431static inline int notify_qemu_malloc(volatile MallocDesc* desc) { 432 desc->libc_pid = malloc_pid; 433 desc->allocator_pid = getpid(); 434 desc->av_count = 0; 435 notify_qemu(TRACE_DEV_REG_MALLOC, (uint32_t)desc); 436 437 /* Emulator reports failure by zeroing libc_pid field of the 438 * descriptor. */ 439 return desc->libc_pid != 0 ? 0 : -1; 440} 441 442/* Fires up TRACE_DEV_REG_FREE_PTR event in the emulator. 443 * Param: 444 * ptr - Pointer to the memory block that's being freed. 445 * Return: 446 * Zero on success, or -1 on failure. 447 */ 448static inline int notify_qemu_free(void* ptr_to_free) { 449 volatile MallocFree free_desc; 450 451 free_desc.ptr = ptr_to_free; 452 free_desc.libc_pid = malloc_pid; 453 free_desc.free_pid = getpid(); 454 notify_qemu(TRACE_DEV_REG_FREE_PTR, (uint32_t)&free_desc); 455 456 /* Emulator reports failure by zeroing libc_pid field of the 457 * descriptor. */ 458 return free_desc.libc_pid != 0 ? 0 : -1; 459} 460 461/* Fires up TRACE_DEV_REG_QUERY_MALLOC event in the emulator. 462 * Param: 463 * ptr - Pointer to request allocation information for. 464 * desc - Pointer to MallocDesc instance that will receive allocation 465 * information. 466 * routine - Code of the allocation routine, in context of which query is made: 467 * 1 - free 468 * 2 - realloc 469 * Return: 470 * Zero on success, or -1 on failure. 471 */ 472static inline int query_qemu_malloc_info(void* ptr, MallocDesc* desc, uint32_t routine) { 473 volatile MallocDescQuery query; 474 475 query.ptr = ptr; 476 query.libc_pid = malloc_pid; 477 query.query_pid = getpid(); 478 query.routine = routine; 479 query.desc = desc; 480 notify_qemu(TRACE_DEV_REG_QUERY_MALLOC, (uint32_t)&query); 481 482 /* Emulator reports failure by zeroing libc_pid field of the 483 * descriptor. */ 484 return query.libc_pid != 0 ? 0 : -1; 485} 486 487/* Logs a message to emulator's stdout. 488 * Param: 489 * prio - Message priority (debug, info, or error) 490 * fmt + rest - Message format and parameters. 491 */ 492static void qemu_log(int prio, const char* fmt, ...) { 493 va_list ap; 494 char buf[4096]; 495 const char* prefix; 496 497 /* Choose message prefix depending on the priority value. */ 498 switch (prio) { 499 case ANDROID_LOG_ERROR: 500 if (!tracing_enabled(error)) { 501 return; 502 } 503 prefix = "E"; 504 break; 505 case ANDROID_LOG_INFO: 506 if (!tracing_enabled(info)) { 507 return; 508 } 509 prefix = "I"; 510 break; 511 case ANDROID_LOG_DEBUG: 512 default: 513 if (!tracing_enabled(debug)) { 514 return; 515 } 516 prefix = "D"; 517 break; 518 } 519 520 va_start(ap, fmt); 521 vsnprintf(buf, sizeof(buf), fmt, ap); 522 va_end(ap); 523 buf[sizeof(buf) - 1] = '\0'; 524 525 TR("%s/memcheck: %s\n", prefix, buf); 526} 527 528/* Dumps content of memory allocation descriptor to a string. 529 * Param: 530 * str - String to dump descriptor to. 531 * str_buf_size - Size of string's buffer. 532 * desc - Descriptor to dump. 533 */ 534static void dump_malloc_descriptor(char* str, size_t str_buf_size, const MallocDesc* desc) { 535 if (str_buf_size) { 536 snprintf(str, str_buf_size, 537 "MDesc: %p: %X <-> %X [%u + %u + %u] by pid=%03u in libc_pid=%03u", 538 mallocdesc_user_ptr(desc), (uint32_t)desc->ptr, 539 (uint32_t)mallocdesc_alloc_end(desc), desc->prefix_size, 540 desc->requested_bytes, desc->suffix_size, desc->allocator_pid, 541 desc->libc_pid); 542 str[str_buf_size - 1] = '\0'; 543 } 544} 545 546#if TEST_ACCESS_VIOLATIONS 547/* Causes an access violation on allocation descriptor, and verifies that 548 * violation has been detected by memory checker in the emulator. 549 */ 550static void test_access_violation(const MallocDesc* desc) { 551 MallocDesc desc_chk; 552 char ch; 553 volatile char* prefix = (volatile char*)desc->ptr; 554 volatile char* suffix = (volatile char*)mallocdesc_user_ptr(desc) + 555 desc->requested_bytes; 556 /* We're causing AV by reading from the prefix and suffix areas of the 557 * allocated block. This should produce two access violations, so when we 558 * get allocation descriptor from QEMU, av_counter should be bigger than 559 * av_counter of the original descriptor by 2. */ 560 ch = *prefix; 561 ch = *suffix; 562 if (!query_qemu_malloc_info(mallocdesc_user_ptr(desc), &desc_chk, 2) && 563 desc_chk.av_count != (desc->av_count + 2)) { 564 log_mdesc(error, &desc_chk, 565 "<libc_pid=%03u, pid=%03u>: malloc: Access violation test failed:\n" 566 "Expected violations count %u is not equal to the actually reported %u", 567 malloc_pid, getpid(), desc->av_count + 2, 568 desc_chk.av_count); 569 } 570} 571#endif // TEST_ACCESS_VIOLATIONS 572 573// ============================================================================= 574// API routines 575// ============================================================================= 576 577void* qemu_instrumented_malloc(size_t bytes); 578void qemu_instrumented_free(void* mem); 579void* qemu_instrumented_calloc(size_t n_elements, size_t elem_size); 580void* qemu_instrumented_realloc(void* mem, size_t bytes); 581void* qemu_instrumented_memalign(size_t alignment, size_t bytes); 582 583/* Initializes malloc debugging instrumentation for the emulator. 584 * This routine is called from malloc_init_impl routine implemented in 585 * bionic/libc/bionic/malloc_debug_common.c when malloc debugging gets 586 * initialized for a process. The way malloc debugging implementation is 587 * done, it is guaranteed that this routine will be called just once per 588 * process. 589 * Return: 590 * 0 on success, or -1 on failure. 591*/ 592int malloc_debug_initialize() { 593 /* We will be using emulator's magic page to report memory allocation 594 * activities. In essence, what magic page does, it translates writes to 595 * the memory mapped spaces into writes to an I/O port that emulator 596 * "listens to" on the other end. Note that until we open and map that 597 * device, logging to emulator's stdout will not be available. */ 598 int fd = open("/dev/qemu_trace", O_RDWR); 599 if (fd < 0) { 600 error_log("Unable to open /dev/qemu_trace"); 601 return -1; 602 } else { 603 qtrace = mmap(NULL, PAGESIZE, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0); 604 close(fd); 605 606 if (qtrace == MAP_FAILED) { 607 qtrace = NULL; 608 error_log("Unable to mmap /dev/qemu_trace"); 609 return -1; 610 } 611 } 612 613 /* Cache pid of the process this library has been initialized for. */ 614 malloc_pid = getpid(); 615 616 return 0; 617} 618 619/* Completes malloc debugging instrumentation for the emulator. 620 * Note that this routine is called after successful return from 621 * malloc_debug_initialize, which means that connection to the emulator via 622 * "magic page" has been established. 623 * Param: 624 * alignment - Alignment requirement set for memiry allocations. 625 * memcheck_param - Emulator's -memcheck option parameters. This string 626 * contains abbreviation for guest events that are enabled for tracing. 627 * Return: 628 * 0 on success, or -1 on failure. 629*/ 630int memcheck_initialize(int alignment, const char* memcheck_param) { 631 malloc_alignment = alignment; 632 633 /* Parse -memcheck parameter for the guest tracing flags. */ 634 while (*memcheck_param != '\0') { 635 switch (*memcheck_param) { 636 case 'a': 637 // Enable all messages from the guest. 638 tracing_flags |= ALL_TRACING_ENABLED; 639 break; 640 case 'd': 641 // Enable debug messages from the guest. 642 tracing_flags |= DEBUG_TRACING_ENABLED; 643 break; 644 case 'e': 645 // Enable error messages from the guest. 646 tracing_flags |= ERROR_TRACING_ENABLED; 647 break; 648 case 'i': 649 // Enable info messages from the guest. 650 tracing_flags |= INFO_TRACING_ENABLED; 651 break; 652 default: 653 break; 654 } 655 if (tracing_flags == ALL_TRACING_ENABLED) { 656 break; 657 } 658 memcheck_param++; 659 } 660 661 notify_qemu_libc_initialized(malloc_pid); 662 663 qemu_debug_log("Instrumented for pid=%03u: malloc=%p, free=%p, calloc=%p, realloc=%p, memalign=%p", 664 malloc_pid, qemu_instrumented_malloc, qemu_instrumented_free, 665 qemu_instrumented_calloc, qemu_instrumented_realloc, 666 qemu_instrumented_memalign); 667 668 return 0; 669} 670 671/* This routine serves as entry point for 'malloc'. 672 * Primary responsibility of this routine is to allocate requested number of 673 * bytes (plus prefix, and suffix guards), and report allocation to the 674 * emulator. 675 */ 676void* qemu_instrumented_malloc(size_t bytes) { 677 MallocDesc desc; 678 679 /* Initialize block descriptor and allocate memory. Note that dlmalloc 680 * returns a valid pointer on zero allocation. Lets mimic this behavior. */ 681 desc.prefix_size = DEFAULT_PREFIX_SIZE; 682 desc.requested_bytes = bytes; 683 desc.suffix_size = DEFAULT_SUFFIX_SIZE; 684 desc.ptr = dlmalloc(mallocdesc_alloc_size(&desc)); 685 if (desc.ptr == NULL) { 686 qemu_error_log("<libc_pid=%03u, pid=%03u> malloc(%u): dlmalloc(%u) failed.", 687 malloc_pid, getpid(), bytes, mallocdesc_alloc_size(&desc)); 688 return NULL; 689 } 690 691 // Fire up event in the emulator. 692 if (notify_qemu_malloc(&desc)) { 693 log_mdesc(error, &desc, "<libc_pid=%03u, pid=%03u>: malloc: notify_malloc failed for ", 694 malloc_pid, getpid()); 695 dlfree(desc.ptr); 696 return NULL; 697 } else { 698#if TEST_ACCESS_VIOLATIONS 699 test_access_violation(&desc); 700#endif // TEST_ACCESS_VIOLATIONS 701 log_mdesc(info, &desc, "+++ <libc_pid=%03u, pid=%03u> malloc(%u) -> ", 702 malloc_pid, getpid(), bytes); 703 return mallocdesc_user_ptr(&desc); 704 } 705} 706 707/* This routine serves as entry point for 'malloc'. 708 * Primary responsibility of this routine is to free requested memory, and 709 * report free block to the emulator. 710 */ 711void qemu_instrumented_free(void* mem) { 712 MallocDesc desc; 713 714 if (mem == NULL) { 715 // Just let go NULL free 716 dlfree(mem); 717 return; 718 } 719 720 // Query emulator for the freeing block information. 721 if (query_qemu_malloc_info(mem, &desc, 1)) { 722 error_log("<libc_pid=%03u, pid=%03u>: free(%p) query_info failed.", 723 malloc_pid, getpid(), mem); 724 return; 725 } 726 727#if TEST_ACCESS_VIOLATIONS 728 test_access_violation(&desc); 729#endif // TEST_ACCESS_VIOLATIONS 730 731 /* Make sure that pointer that's being freed matches what we expect 732 * for this memory block. Note that this violation should be already 733 * caught in the emulator. */ 734 if (mem != mallocdesc_user_ptr(&desc)) { 735 log_mdesc(error, &desc, "<libc_pid=%03u, pid=%03u>: free(%p) is invalid for ", 736 malloc_pid, getpid(), mem); 737 return; 738 } 739 740 // Fire up event in the emulator and free block that was actually allocated. 741 if (notify_qemu_free(mem)) { 742 log_mdesc(error, &desc, "<libc_pid=%03u, pid=%03u>: free(%p) notify_free failed for ", 743 malloc_pid, getpid(), mem); 744 } else { 745 log_mdesc(info, &desc, "--- <libc_pid=%03u, pid=%03u> free(%p) -> ", 746 malloc_pid, getpid(), mem); 747 dlfree(desc.ptr); 748 } 749} 750 751/* This routine serves as entry point for 'calloc'. 752 * This routine behaves similarly to qemu_instrumented_malloc. 753 */ 754void* qemu_instrumented_calloc(size_t n_elements, size_t elem_size) { 755 if (n_elements == 0 || elem_size == 0) { 756 // Just let go zero bytes allocation. 757 qemu_info_log("::: <libc_pid=%03u, pid=%03u>: Zero calloc redir to malloc", 758 malloc_pid, getpid()); 759 return qemu_instrumented_malloc(0); 760 } 761 762 /* Fail on overflow - just to be safe even though this code runs only 763 * within the debugging C library, not the production one */ 764 if (n_elements && MAX_SIZE_T / n_elements < elem_size) { 765 return NULL; 766 } 767 768 MallocDesc desc; 769 770 /* Calculating prefix size. The trick here is to make sure that 771 * first element (returned to the caller) is properly aligned. */ 772 if (DEFAULT_PREFIX_SIZE >= elem_size) { 773 /* If default alignment is bigger than element size, we will 774 * set our prefix size to the default alignment size. */ 775 desc.prefix_size = DEFAULT_PREFIX_SIZE; 776 /* For the suffix we will use whatever bytes remain from the prefix 777 * allocation size, aligned to the size of an element, plus the usual 778 * default suffix size. */ 779 desc.suffix_size = (DEFAULT_PREFIX_SIZE % elem_size) + 780 DEFAULT_SUFFIX_SIZE; 781 } else { 782 /* Make sure that prefix, and suffix sizes is at least elem_size, 783 * and first element returned to the caller is properly aligned. */ 784 desc.prefix_size = elem_size + DEFAULT_PREFIX_SIZE - 1; 785 desc.prefix_size &= ~(malloc_alignment - 1); 786 desc.suffix_size = DEFAULT_SUFFIX_SIZE; 787 } 788 desc.requested_bytes = n_elements * elem_size; 789 size_t total_size = desc.requested_bytes + desc.prefix_size + desc.suffix_size; 790 size_t total_elements = total_size / elem_size; 791 total_size %= elem_size; 792 if (total_size != 0) { 793 // Add extra to the suffix area. 794 total_elements++; 795 desc.suffix_size += (elem_size - total_size); 796 } 797 desc.ptr = dlcalloc(total_elements, elem_size); 798 if (desc.ptr == NULL) { 799 error_log("<libc_pid=%03u, pid=%03u> calloc: dlcalloc(%u(%u), %u) (prx=%u, sfx=%u) failed.", 800 malloc_pid, getpid(), n_elements, total_elements, elem_size, 801 desc.prefix_size, desc.suffix_size); 802 return NULL; 803 } 804 805 if (notify_qemu_malloc(&desc)) { 806 log_mdesc(error, &desc, "<libc_pid=%03u, pid=%03u>: calloc(%u(%u), %u): notify_malloc failed for ", 807 malloc_pid, getpid(), n_elements, total_elements, elem_size); 808 dlfree(desc.ptr); 809 return NULL; 810 } else { 811#if TEST_ACCESS_VIOLATIONS 812 test_access_violation(&desc); 813#endif // TEST_ACCESS_VIOLATIONS 814 log_mdesc(info, &desc, "### <libc_pid=%03u, pid=%03u> calloc(%u(%u), %u) -> ", 815 malloc_pid, getpid(), n_elements, total_elements, elem_size); 816 return mallocdesc_user_ptr(&desc); 817 } 818} 819 820/* This routine serves as entry point for 'realloc'. 821 * This routine behaves similarly to qemu_instrumented_free + 822 * qemu_instrumented_malloc. Note that this modifies behavior of "shrinking" an 823 * allocation, but overall it doesn't seem to matter, as caller of realloc 824 * should not expect that pointer returned after shrinking will remain the same. 825 */ 826void* qemu_instrumented_realloc(void* mem, size_t bytes) { 827 MallocDesc new_desc; 828 MallocDesc cur_desc; 829 size_t to_copy; 830 void* ret; 831 832 if (mem == NULL) { 833 // Nothing to realloc. just do regular malloc. 834 qemu_info_log("::: <libc_pid=%03u, pid=%03u>: realloc(%p, %u) redir to malloc", 835 malloc_pid, getpid(), mem, bytes); 836 return qemu_instrumented_malloc(bytes); 837 } 838 839 if (bytes == 0) { 840 // This is a "free" condition. 841 qemu_info_log("::: <libc_pid=%03u, pid=%03u>: realloc(%p, %u) redir to free and malloc", 842 malloc_pid, getpid(), mem, bytes); 843 qemu_instrumented_free(mem); 844 845 // This is what dlrealloc does for a "free" realloc. 846 return NULL; 847 } 848 849 // Query emulator for the reallocating block information. 850 if (query_qemu_malloc_info(mem, &cur_desc, 2)) { 851 // Note that this violation should be already caught in the emulator. 852 error_log("<libc_pid=%03u, pid=%03u>: realloc(%p, %u) query_info failed.", 853 malloc_pid, getpid(), mem, bytes); 854 return NULL; 855 } 856 857#if TEST_ACCESS_VIOLATIONS 858 test_access_violation(&cur_desc); 859#endif // TEST_ACCESS_VIOLATIONS 860 861 /* Make sure that reallocating pointer value is what we would expect 862 * for this memory block. Note that this violation should be already caught 863 * in the emulator.*/ 864 if (mem != mallocdesc_user_ptr(&cur_desc)) { 865 log_mdesc(error, &cur_desc, "<libc_pid=%03u, pid=%03u>: realloc(%p, %u) is invalid for ", 866 malloc_pid, getpid(), mem, bytes); 867 return NULL; 868 } 869 870 /* TODO: We're a bit inefficient here, always allocating new block from 871 * the heap. If this realloc shrinks current buffer, we can just do the 872 * shrinking "in place", adjusting suffix_size in the allocation descriptor 873 * for this block that is stored in the emulator. */ 874 875 // Initialize descriptor for the new block. 876 new_desc.prefix_size = DEFAULT_PREFIX_SIZE; 877 new_desc.requested_bytes = bytes; 878 new_desc.suffix_size = DEFAULT_SUFFIX_SIZE; 879 new_desc.ptr = dlmalloc(mallocdesc_alloc_size(&new_desc)); 880 if (new_desc.ptr == NULL) { 881 log_mdesc(error, &cur_desc, "<libc_pid=%03u, pid=%03u>: realloc(%p, %u): dlmalloc(%u) failed on ", 882 malloc_pid, getpid(), mem, bytes, 883 mallocdesc_alloc_size(&new_desc)); 884 return NULL; 885 } 886 ret = mallocdesc_user_ptr(&new_desc); 887 888 // Copy user data from old block to the new one. 889 to_copy = bytes < cur_desc.requested_bytes ? bytes : 890 cur_desc.requested_bytes; 891 if (to_copy != 0) { 892 memcpy(ret, mallocdesc_user_ptr(&cur_desc), to_copy); 893 } 894 895 // Register new block with emulator. 896 if (notify_qemu_malloc(&new_desc)) { 897 log_mdesc(error, &new_desc, "<libc_pid=%03u, pid=%03u>: realloc(%p, %u) notify_malloc failed -> ", 898 malloc_pid, getpid(), mem, bytes); 899 log_mdesc(error, &cur_desc, " <- "); 900 dlfree(new_desc.ptr); 901 return NULL; 902 } 903 904#if TEST_ACCESS_VIOLATIONS 905 test_access_violation(&new_desc); 906#endif // TEST_ACCESS_VIOLATIONS 907 908 // Free old block. 909 if (notify_qemu_free(mem)) { 910 log_mdesc(error, &cur_desc, "<libc_pid=%03u, pid=%03u>: realloc(%p, %u): notify_free failed for ", 911 malloc_pid, getpid(), mem, bytes); 912 /* Since we registered new decriptor with the emulator, we need 913 * to unregister it before freeing newly allocated block. */ 914 notify_qemu_free(mallocdesc_user_ptr(&new_desc)); 915 dlfree(new_desc.ptr); 916 return NULL; 917 } 918 dlfree(cur_desc.ptr); 919 920 log_mdesc(info, &new_desc, "=== <libc_pid=%03u, pid=%03u>: realloc(%p, %u) -> ", 921 malloc_pid, getpid(), mem, bytes); 922 log_mdesc(info, &cur_desc, " <- "); 923 924 return ret; 925} 926 927/* This routine serves as entry point for 'memalign'. 928 * This routine behaves similarly to qemu_instrumented_malloc. 929 */ 930void* qemu_instrumented_memalign(size_t alignment, size_t bytes) { 931 MallocDesc desc; 932 933 if (bytes == 0) { 934 // Just let go zero bytes allocation. 935 qemu_info_log("::: <libc_pid=%03u, pid=%03u>: memalign(%X, %u) redir to malloc", 936 malloc_pid, getpid(), alignment, bytes); 937 return qemu_instrumented_malloc(0); 938 } 939 940 /* Prefix size for aligned allocation must be equal to the alignment used 941 * for allocation in order to ensure proper alignment of the returned 942 * pointer, in case that alignment requirement is greater than prefix 943 * size. */ 944 desc.prefix_size = alignment > DEFAULT_PREFIX_SIZE ? alignment : 945 DEFAULT_PREFIX_SIZE; 946 desc.requested_bytes = bytes; 947 desc.suffix_size = DEFAULT_SUFFIX_SIZE; 948 desc.ptr = dlmemalign(desc.prefix_size, mallocdesc_alloc_size(&desc)); 949 if (desc.ptr == NULL) { 950 error_log("<libc_pid=%03u, pid=%03u> memalign(%X, %u): dlmalloc(%u) failed.", 951 malloc_pid, getpid(), alignment, bytes, 952 mallocdesc_alloc_size(&desc)); 953 return NULL; 954 } 955 if (notify_qemu_malloc(&desc)) { 956 log_mdesc(error, &desc, "<libc_pid=%03u, pid=%03u>: memalign(%X, %u): notify_malloc failed for ", 957 malloc_pid, getpid(), alignment, bytes); 958 dlfree(desc.ptr); 959 return NULL; 960 } 961 962#if TEST_ACCESS_VIOLATIONS 963 test_access_violation(&desc); 964#endif // TEST_ACCESS_VIOLATIONS 965 966 log_mdesc(info, &desc, "@@@ <libc_pid=%03u, pid=%03u> memalign(%X, %u) -> ", 967 malloc_pid, getpid(), alignment, bytes); 968 return mallocdesc_user_ptr(&desc); 969} 970