1 /* 2 * Copyright (C) 2015 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 #ifndef DEBUG_MALLOC_TRACKDATA_H 30 #define DEBUG_MALLOC_TRACKDATA_H 31 32 #include <stdint.h> 33 #include <pthread.h> 34 35 #include <vector> 36 #include <unordered_set> 37 38 #include <private/bionic_macros.h> 39 40 // Forward declarations. 41 struct Header; 42 struct Config; 43 class DebugData; 44 45 class TrackData { 46 public: 47 TrackData() = default; 48 virtual ~TrackData() = default; 49 50 void GetList(std::vector<const Header*>* list); 51 52 void Add(const Header* header, bool backtrace_found); 53 54 void Remove(const Header* header, bool backtrace_found); 55 56 bool Contains(const Header *header); 57 58 void GetInfo(DebugData& debug, uint8_t** info, size_t* overall_size, 59 size_t* info_size, size_t* total_memory, size_t* backtrace_size); 60 61 void DisplayLeaks(DebugData& debug); 62 PrepareFork()63 void PrepareFork() { pthread_mutex_lock(&mutex_); } PostForkParent()64 void PostForkParent() { pthread_mutex_unlock(&mutex_); } PostForkChild()65 void PostForkChild() { pthread_mutex_init(&mutex_, NULL); } 66 67 private: 68 pthread_mutex_t mutex_ = PTHREAD_MUTEX_INITIALIZER; 69 std::unordered_set<const Header*> headers_; 70 size_t total_backtrace_allocs_ = 0; 71 72 DISALLOW_COPY_AND_ASSIGN(TrackData); 73 }; 74 75 #endif // DEBUG_MALLOC_TRACKDATA_H 76