platform_bionic/libc/malloc_debug/DebugData.h
Colin Cross 7a28a3cf1f malloc_debug: reset TrackData mutex after fork
Add a pthread_atfork handler to malloc_debug to lock the TrackData mutex
during fork and reset it in the child.  Ensures that the TrackData is
consistent when forking from a multi-threaded process, and that the
mutex is in a defined state in the child.

Change-Id: I0dfed30045a28551217ceac227d33b2e18894932
2016-02-12 16:41:14 -08:00

106 lines
3.4 KiB
C++

/*
* Copyright (C) 2015 The Android Open Source Project
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* * Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in
* the documentation and/or other materials provided with the
* distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
* FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
* COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
* OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
* AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
* OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/
#ifndef DEBUG_MALLOC_DEBUGDATA_H
#define DEBUG_MALLOC_DEBUGDATA_H
#include <stdint.h>
#include <memory>
#include <vector>
#include <private/bionic_macros.h>
#include "BacktraceData.h"
#include "Config.h"
#include "FreeTrackData.h"
#include "GuardData.h"
#include "malloc_debug.h"
#include "TrackData.h"
class DebugData {
public:
DebugData() = default;
~DebugData() = default;
bool Initialize();
static bool Disabled();
inline void* GetPointer(const Header* header) {
uintptr_t value = reinterpret_cast<uintptr_t>(header);
return reinterpret_cast<void*>(value + pointer_offset_);
}
Header* GetHeader(const void* pointer) {
uintptr_t value = reinterpret_cast<uintptr_t>(pointer);
return reinterpret_cast<Header*>(value - pointer_offset_);
}
BacktraceHeader* GetAllocBacktrace(const Header* header) {
uintptr_t value = reinterpret_cast<uintptr_t>(header);
return reinterpret_cast<BacktraceHeader*>(value + backtrace->alloc_offset());
}
uint8_t* GetFrontGuard(const Header* header) {
uintptr_t value = reinterpret_cast<uintptr_t>(header);
return reinterpret_cast<uint8_t*>(value + front_guard->offset());
}
uint8_t* GetRearGuard(const Header* header) {
uintptr_t value = reinterpret_cast<uintptr_t>(GetPointer(header));
return reinterpret_cast<uint8_t*>(value + header->real_size());
}
const Config& config() { return config_; }
size_t pointer_offset() { return pointer_offset_; }
bool need_header() { return need_header_; }
size_t extra_bytes() { return extra_bytes_; }
void PrepareFork();
void PostForkParent();
void PostForkChild();
std::unique_ptr<BacktraceData> backtrace;
std::unique_ptr<TrackData> track;
std::unique_ptr<FrontGuardData> front_guard;
std::unique_ptr<RearGuardData> rear_guard;
std::unique_ptr<FreeTrackData> free_track;
private:
size_t extra_bytes_ = 0;
size_t pointer_offset_ = 0;
bool need_header_ = false;
Config config_;
DISALLOW_COPY_AND_ASSIGN(DebugData);
};
#endif // MALLOC_DEBUG_DEBUGDATA_H