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