1/*
2 *  Copyright (C) 2008 Apple Inc. All rights reserved.
3 *
4 *  This library is free software; you can redistribute it and/or
5 *  modify it under the terms of the GNU Library General Public
6 *  License as published by the Free Software Foundation; either
7 *  version 2 of the License, or (at your option) any later version.
8 *
9 *  This library is distributed in the hope that it will be useful,
10 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
11 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12 *  Library General Public License for more details.
13 *
14 *  You should have received a copy of the GNU Library General Public License
15 *  along with this library; see the file COPYING.LIB.  If not, write to
16 *  the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
17 *  Boston, MA 02110-1301, USA.
18 *
19 */
20
21#include "config.h"
22#include "RefCountedLeakCounter.h"
23
24#include <wtf/HashCountedSet.h>
25
26namespace WTF {
27
28#ifdef NDEBUG
29
30void RefCountedLeakCounter::suppressMessages(const char*) { }
31void RefCountedLeakCounter::cancelMessageSuppression(const char*) { }
32
33RefCountedLeakCounter::RefCountedLeakCounter(const char*) { }
34RefCountedLeakCounter::~RefCountedLeakCounter() { }
35
36void RefCountedLeakCounter::increment() { }
37void RefCountedLeakCounter::decrement() { }
38
39#else
40
41#define LOG_CHANNEL_PREFIX Log
42static WTFLogChannel LogRefCountedLeaks = { WTFLogChannelOn, "RefCountedLeaks" };
43
44typedef HashCountedSet<const char*, PtrHash<const char*>> ReasonSet;
45static ReasonSet* leakMessageSuppressionReasons;
46
47void RefCountedLeakCounter::suppressMessages(const char* reason)
48{
49    if (!leakMessageSuppressionReasons)
50        leakMessageSuppressionReasons = new ReasonSet;
51    leakMessageSuppressionReasons->add(reason);
52}
53
54void RefCountedLeakCounter::cancelMessageSuppression(const char* reason)
55{
56    ASSERT(leakMessageSuppressionReasons);
57    ASSERT(leakMessageSuppressionReasons->contains(reason));
58    leakMessageSuppressionReasons->remove(reason);
59}
60
61RefCountedLeakCounter::RefCountedLeakCounter(const char* description)
62    : m_description(description)
63{
64}
65
66RefCountedLeakCounter::~RefCountedLeakCounter()
67{
68    static bool loggedSuppressionReason;
69    if (m_count) {
70        if (!leakMessageSuppressionReasons || leakMessageSuppressionReasons->isEmpty())
71            LOG(RefCountedLeaks, "LEAK: %u %s", m_count.load(), m_description);
72        else if (!loggedSuppressionReason) {
73            // This logs only one reason. Later we could change it so we log all the reasons.
74            LOG(RefCountedLeaks, "No leak checking done: %s", leakMessageSuppressionReasons->begin()->key);
75            loggedSuppressionReason = true;
76        }
77    }
78}
79
80void RefCountedLeakCounter::increment()
81{
82    ++m_count;
83}
84
85void RefCountedLeakCounter::decrement()
86{
87    --m_count;
88}
89
90#endif
91
92} // namespace WTF
93