OLD | NEW |
1 /* | 1 // Copyright 2017 The Chromium Authors. All rights reserved. |
2 * Copyright (C) 2007, 2008, 2010 Apple Inc. All rights reserved. | 2 // Use of this source code is governed by a BSD-style license that can be |
3 * Copyright (C) 2007 Justin Haygood (jhaygood@reaktix.com) | 3 // found in the LICENSE file. |
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 * | |
9 * 1. Redistributions of source code must retain the above copyright | |
10 * notice, this list of conditions and the following disclaimer. | |
11 * 2. Redistributions in binary form must reproduce the above copyright | |
12 * notice, this list of conditions and the following disclaimer in the | |
13 * documentation and/or other materials provided with the distribution. | |
14 * 3. Neither the name of Apple Computer, Inc. ("Apple") nor the names of | |
15 * its contributors may be used to endorse or promote products derived | |
16 * from this software without specific prior written permission. | |
17 * | |
18 * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY | |
19 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED | |
20 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE | |
21 * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY | |
22 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES | |
23 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; | |
24 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND | |
25 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | |
26 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF | |
27 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |
28 */ | |
29 | 4 |
30 #ifndef ThreadSafeRefCounted_h | 5 #include "platform/wtf/ThreadSafeRefCounted.h" |
31 #define ThreadSafeRefCounted_h | |
32 | 6 |
33 #include "wtf/Allocator.h" | 7 // The contents of this header was moved to platform/wtf as part of |
34 #include "wtf/Atomics.h" | 8 // WTF migration project. See the following post for details: |
35 #include "wtf/DynamicAnnotations.h" | 9 // https://groups.google.com/a/chromium.org/d/msg/blink-dev/tLdAZCTlcAA/bYXVT8gY
CAAJ |
36 #include "wtf/Noncopyable.h" | |
37 #include "wtf/WTFExport.h" | |
38 | |
39 namespace WTF { | |
40 | |
41 class WTF_EXPORT ThreadSafeRefCountedBase { | |
42 WTF_MAKE_NONCOPYABLE(ThreadSafeRefCountedBase); | |
43 USING_FAST_MALLOC(ThreadSafeRefCountedBase); | |
44 | |
45 public: | |
46 ThreadSafeRefCountedBase(int initialRefCount = 1) | |
47 : m_refCount(initialRefCount) {} | |
48 | |
49 void ref() { atomicIncrement(&m_refCount); } | |
50 | |
51 bool hasOneRef() { return refCount() == 1; } | |
52 | |
53 int refCount() const { return static_cast<int const volatile&>(m_refCount); } | |
54 | |
55 protected: | |
56 // Returns whether the pointer should be freed or not. | |
57 bool derefBase() { | |
58 WTF_ANNOTATE_HAPPENS_BEFORE(&m_refCount); | |
59 if (atomicDecrement(&m_refCount) <= 0) { | |
60 WTF_ANNOTATE_HAPPENS_AFTER(&m_refCount); | |
61 return true; | |
62 } | |
63 return false; | |
64 } | |
65 | |
66 private: | |
67 int m_refCount; | |
68 }; | |
69 | |
70 template <class T> | |
71 class ThreadSafeRefCounted : public ThreadSafeRefCountedBase { | |
72 public: | |
73 void deref() { | |
74 if (derefBase()) | |
75 delete static_cast<T*>(this); | |
76 } | |
77 | |
78 protected: | |
79 ThreadSafeRefCounted() {} | |
80 }; | |
81 | |
82 } // namespace WTF | |
83 | |
84 using WTF::ThreadSafeRefCounted; | |
85 | |
86 #endif // ThreadSafeRefCounted_h | |
OLD | NEW |