kakoune/src/safe_ptr.hh

166 lines
3.7 KiB
C++
Raw Normal View History

#ifndef safe_ptr_hh_INCLUDED
#define safe_ptr_hh_INCLUDED
// #define SAFE_PTR_TRACK_CALLSTACKS
#include "assert.hh"
#ifdef SAFE_PTR_TRACK_CALLSTACKS
2015-01-12 14:58:41 +01:00
#include "vector.hh"
#include <algorithm>
#endif
namespace Kakoune
{
// *** SafePtr: objects that assert nobody references them when they die ***
template<typename T>
class SafePtr
{
public:
SafePtr() : m_ptr(nullptr) {}
explicit SafePtr(T* ptr) : m_ptr(ptr)
{
#ifdef KAK_DEBUG
if (m_ptr)
m_ptr->inc_safe_count(this);
#endif
}
SafePtr(const SafePtr& other) : SafePtr(other.m_ptr) {}
SafePtr(SafePtr&& other) noexcept : m_ptr(other.m_ptr)
{
other.m_ptr = nullptr;
#ifdef KAK_DEBUG
if (m_ptr)
m_ptr->safe_ptr_moved(&other, this);
#endif
}
~SafePtr()
{
#ifdef KAK_DEBUG
if (m_ptr)
m_ptr->dec_safe_count(this);
#endif
}
SafePtr& operator=(const SafePtr& other)
{
#ifdef KAK_DEBUG
if (m_ptr != other.m_ptr)
{
if (m_ptr)
m_ptr->dec_safe_count(this);
if (other.m_ptr)
other.m_ptr->inc_safe_count(this);
}
#endif
m_ptr = other.m_ptr;
return *this;
}
SafePtr& operator=(SafePtr&& other) noexcept
{
#ifdef KAK_DEBUG
if (m_ptr)
m_ptr->dec_safe_count(this);
if (other.m_ptr)
other.m_ptr->safe_ptr_moved(&other, this);
#endif
m_ptr = other.m_ptr;
other.m_ptr = nullptr;
return *this;
}
void reset(T* ptr = nullptr)
{
*this = SafePtr(ptr);
}
bool operator== (const SafePtr& other) const { return m_ptr == other.m_ptr; }
bool operator!= (const SafePtr& other) const { return m_ptr != other.m_ptr; }
bool operator== (T* ptr) const { return m_ptr == ptr; }
bool operator!= (T* ptr) const { return m_ptr != ptr; }
T& operator* () const { return *m_ptr; }
T* operator-> () const { return m_ptr; }
T* get() const { return m_ptr; }
explicit operator bool() const { return m_ptr; }
private:
T* m_ptr;
};
class SafeCountable
{
public:
#ifdef KAK_DEBUG
SafeCountable() : m_count(0) {}
~SafeCountable()
{
kak_assert(m_count == 0);
#ifdef SAFE_PTR_TRACK_CALLSTACKS
kak_assert(m_callstacks.empty());
#endif
}
void inc_safe_count(void* ptr) const
{
++m_count;
#ifdef SAFE_PTR_TRACK_CALLSTACKS
m_callstacks.emplace_back(ptr);
#endif
}
void dec_safe_count(void* ptr) const
{
--m_count;
kak_assert(m_count >= 0);
#ifdef SAFE_PTR_TRACK_CALLSTACKS
auto it = std::find_if(m_callstacks.begin(), m_callstacks.end(),
[=](const Callstack& cs) { return cs.ptr == ptr; });
kak_assert(it != m_callstacks.end());
m_callstacks.erase(it);
#endif
}
void safe_ptr_moved(void* from, void* to) const
{
#ifdef SAFE_PTR_TRACK_CALLSTACKS
auto it = std::find_if(m_callstacks.begin(), m_callstacks.end(),
[=](const Callstack& cs) { return cs.ptr == from; });
kak_assert(it != m_callstacks.end());
it->ptr = to;
#endif
}
private:
#ifdef SAFE_PTR_TRACK_CALLSTACKS
struct Backtrace
{
static constexpr int max_frames = 16;
void* stackframes[max_frames];
int num_frames = 0;
Backtrace();
const char* desc() const;
};
struct Callstack
{
Callstack(void* p) : ptr(p) {}
void* ptr;
Backtrace bt;
};
2015-01-12 14:58:41 +01:00
mutable Vector<Callstack> m_callstacks;
#endif
mutable int m_count;
#endif
};
}
#endif // safe_ptr_hh_INCLUDED