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. 41struct Header; 42struct Config; 43class DebugData; 44 45class 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 63 void PrepareFork() { pthread_mutex_lock(&mutex_); } 64 void PostForkParent() { pthread_mutex_unlock(&mutex_); } 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