blob: 2c03c409d97e8a5e07036ae461fec6e0e7205102 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
//===------------------SharedCluster.h --------------------------*- C++ -*-===//
//
// The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#ifndef utility_SharedCluster_h_
#define utility_SharedCluster_h_
#include "lldb/Utility/SharingPtr.h"
#include "lldb/Host/Mutex.h"
#include "llvm/ADT/SmallPtrSet.h"
namespace lldb_private {
namespace imp
{
template <typename T>
class shared_ptr_refcount : public lldb_private::imp::shared_count
{
public:
template<class Y> shared_ptr_refcount (Y *in) : shared_count (0), manager(in) {}
shared_ptr_refcount() : shared_count (0) {}
~shared_ptr_refcount() override
{
}
void on_zero_shared() override
{
manager->DecrementRefCount();
}
private:
T *manager;
};
} // namespace imp
template <class T>
class ClusterManager
{
public:
ClusterManager () :
m_objects(),
m_external_ref(0),
m_mutex(Mutex::eMutexTypeNormal) {}
~ClusterManager ()
{
for (typename llvm::SmallPtrSet<T *, 16>::iterator pos = m_objects.begin(), end = m_objects.end(); pos != end; ++pos)
{
T *object = *pos;
delete object;
}
// Decrement refcount should have been called on this ClusterManager,
// and it should have locked the mutex, now we will unlock it before
// we destroy it...
m_mutex.Unlock();
}
void ManageObject (T *new_object)
{
Mutex::Locker locker (m_mutex);
m_objects.insert (new_object);
}
typename lldb_private::SharingPtr<T> GetSharedPointer(T *desired_object)
{
{
Mutex::Locker locker (m_mutex);
m_external_ref++;
assert (m_objects.count(desired_object));
}
return typename lldb_private::SharingPtr<T> (desired_object, new imp::shared_ptr_refcount<ClusterManager> (this));
}
private:
void DecrementRefCount ()
{
m_mutex.Lock();
m_external_ref--;
if (m_external_ref == 0)
delete this;
else
m_mutex.Unlock();
}
friend class imp::shared_ptr_refcount<ClusterManager>;
llvm::SmallPtrSet<T *, 16> m_objects;
int m_external_ref;
Mutex m_mutex;
};
} // namespace lldb_private
#endif // utility_SharedCluster_h_
|
