0
0
Fork 0
mirror of https://github.com/matrix-construct/construct synced 2024-11-30 10:42:47 +01:00
construct/include/ircd/life_guard.h
Jason Volk d9a4f06bf6 ircd: Employ namespace scope extensions from c++1z/gnu++14.
This is not a move to c++17. If the compiler happens to have support
for c++17 namespace scope resolution, they have been kind enough to
backport it to gnu++14. This limits our support for really old c++14
compilers, but that was limited anyway. GCC 6.1 and clang 3.6 tested.
2017-09-08 03:47:46 -07:00

133 lines
4.2 KiB
C++

/*
* Copyright (C) 2016 Charybdis Development Team
* Copyright (C) 2016 Jason Volk <jason@zemos.net>
*
* Permission to use, copy, modify, and/or distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice is present in all copies.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT,
* INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
* STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
* IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
*/
#pragma once
#define HAVE_IRCD_LIFE_GUARD_H
//
// life_guard is a convenience which takes advantage of
// std::enable_shared_from_this<T>. The life_guard glorifies the constructor
// of an std::shared_ptr<T> by accepting std::weak_ptr<T> and T& itself all
// with proper semantics. Once construction is successful, the user holds it
// for the duration of the scope ensuring T& survives context interleaving
// without being destructed.
//
namespace ircd
{
// Tests if type inherits from std::enable_shared_from_this<>
template<class T>
constexpr typename std::enable_if<is_complete<T>::value, bool>::type
is_shared_from_this()
{
return std::is_base_of<std::enable_shared_from_this<T>, T>();
}
// Unconditional failure for fwd-declared incomplete types, which
// obviously don't inherit from std::enable_shared_from_this<>
template<class T>
constexpr typename std::enable_if<!is_complete<T>::value, bool>::type
is_shared_from_this()
{
return false;
}
// Convenience functions for types shared_from_this
template<class T> std::shared_ptr<const T> shared_from(const T &t);
template<class T> std::shared_ptr<T> shared_from(T &t);
template<class T> std::weak_ptr<const T> weak_from(const T &t);
template<class T> std::weak_ptr<T> weak_from(T &t);
template<class T> struct life_guard;
}
/* Use the life_guard to keep an object alive within a function running in a context.
*
* Example:
*
void foo(client &c)
{
const life_guard<client> lg(c);
c.call(); // This call was always safe with or w/o life_guard.
ctx::wait(); // The context has now yielded and another context might destroy client &c
c.call(); // The context continues and this would have made a call on a dead c.
}
*/
template<class T>
struct ircd::life_guard
:std::shared_ptr<T>
{
// This constructor is used when the templated type inherits from std::enable_shared_from_this<>
template<class SFINAE = T>
life_guard(T &t,
typename std::enable_if<is_shared_from_this<SFINAE>(), void>::type * = 0)
:std::shared_ptr<T>(shared_from(t))
{}
// This constructor uses our convention for forward declaring objects that internally
// inherit from std::enable_shared_from_this<>. Our convention is to provide:
//
// std::shared_ptr<T> shared_from(T &c);
//
template<class SFINAE = T>
life_guard(T &t,
typename std::enable_if<!is_shared_from_this<SFINAE>(), void>::type * = 0)
:std::shared_ptr<T>(shared_from(t))
{}
// This constructor is used with a weak_ptr of the type. This throws an exception
// to abort the scope when the object already died before being able to guard at all.
life_guard(const std::weak_ptr<T> &wp)
:std::shared_ptr<T>(wp.lock())
{
if(wp.expired())
throw std::bad_weak_ptr();
}
};
template<class T>
std::weak_ptr<T>
ircd::weak_from(T &t)
{
return shared_from(t);
};
template<class T>
std::weak_ptr<const T>
ircd::weak_from(const T &t)
{
return shared_from(t);
};
template<class T>
std::shared_ptr<T>
ircd::shared_from(T &t)
{
return dynamic_pointer_cast<T>(t.shared_from_this());
};
template<class T>
std::shared_ptr<const T>
ircd::shared_from(const T &t)
{
return dynamic_pointer_cast<const T>(t.shared_from_this());
};