third_party/WebKit/Source/platform/PartitionAllocMemoryDumpProvider.h

// Copyright 2015 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef PartitionAllocMemoryDumpProvider_h
#define PartitionAllocMemoryDumpProvider_h

#include "base/trace_event/memory_dump_provider.h"
#include "public/platform/WebCommon.h"
#include "wtf/Noncopyable.h"
#include "wtf/ThreadingPrimitives.h"

namespace base {
namespace trace_event {

class AllocationRegister;

} // namespace trace_event
} // namespace base

namespace blink {

class BLINK_PLATFORM_EXPORT PartitionAllocMemoryDumpProvider final : public base::trace_event::MemoryDumpProvider {
    // TODO(tasak): PartitionAllocMemoryDumpProvider should be
    // USING_FAST_MALLOC. c.f. crbug.com/584196
    WTF_MAKE_NONCOPYABLE(PartitionAllocMemoryDumpProvider);
public:
    static PartitionAllocMemoryDumpProvider* instance();
    ~PartitionAllocMemoryDumpProvider() override;

    // MemoryDumpProvider implementation.
    bool OnMemoryDump(const base::trace_event::MemoryDumpArgs&, base::trace_event::ProcessMemoryDump*) override;
    void OnHeapProfilingEnabled(bool) override;

    // These methods are called only from PartitionAllocHooks' callbacks.
    void insert(void*, size_t, const char*);
    void remove(void*);

private:
    PartitionAllocMemoryDumpProvider();

    Mutex m_allocationRegisterMutex;
    std::unique_ptr<base::trace_event::AllocationRegister> m_allocationRegister;
    bool m_isHeapProfilingEnabled;
};

} // namespace blink

#endif // PartitionAllocMemoryDumpProvider_h