util.h

Go to the documentation of this file.

// Copyright (c) The Bitcoin Core developers
// Distributed under the MIT software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
 
#ifndef MP_UTIL_H
#define MP_UTIL_H
 
#include <array>
#include <capnp/schema.h>
#include <cassert>
#include <cstdlib>
#include <cstring>
#include <exception>
#include <functional>
#include <kj/async-io.h>
#include <kj/memory.h>
#include <kj/string-tree.h>
#include <mutex>
#include <string>
#include <tuple>
#include <typeinfo>
#include <type_traits>
#include <utility>
#include <variant>
#include <vector>
 
#if __has_include(<cxxabi.h>)
#include <cxxabi.h>
#include <memory>
#endif
 
namespace mp {
 
 
// std::cmp_less_equal/cmp_greater_equal only accept integer types (C++20 §[utility.intcmp]).
// These wrappers fall back to regular comparison for floating-point types.
template <typename A, typename B>
constexpr bool safe_less_equal(A a, B b)
{
    if constexpr (std::is_floating_point_v<A> || std::is_floating_point_v<B>) return a <= b;
    else return std::cmp_less_equal(a, b);
}
template <typename A, typename B>
constexpr bool safe_greater_equal(A a, B b)
{
    if constexpr (std::is_floating_point_v<A> || std::is_floating_point_v<B>) return a >= b;
    else return std::cmp_greater_equal(a, b);
}
 
template <typename... Types>
struct TypeList
{
    static constexpr size_t size = sizeof...(Types);
};
 
template <template <typename...> class Class, typename... Types, typename... Args>
Class<Types..., std::remove_reference_t<Args>...> Make(Args&&... args)
{
    return Class<Types..., std::remove_reference_t<Args>...>{std::forward<Args>(args)...};
}
 
template <std::size_t index, typename List, typename _First = TypeList<>, bool done = index == 0>
struct Split;
 
template <typename _Second, typename _First>
struct Split<0, _Second, _First, true>
{
    using First = _First;
    using Second = _Second;
};
 
template <std::size_t index, typename Type, typename... _Second, typename... _First>
struct Split<index, TypeList<Type, _Second...>, TypeList<_First...>, false>
{
    using _Next = Split<index - 1, TypeList<_Second...>, TypeList<_First..., Type>>;
    using First = typename _Next::First;
    using Second = typename _Next::Second;
};
 
template <typename Callable>
using ResultOf = decltype(std::declval<Callable>()());
 
template <typename T>
using RemoveCvRef = std::remove_cv_t<std::remove_reference_t<T>>;
 
template <typename T>
using Decay = std::decay_t<T>;
 
template <typename T, typename U>
concept FieldTypeIs = std::is_same_v<decltype(std::declval<T>().get()), U>;
 
template <typename T>
concept InterfaceField = requires { typename Decay<decltype(std::declval<T>().get())>::Calls; };
 
template <typename SfinaeExpr, typename Result_>
struct _Require
{
    using Result = Result_;
};
 
template <typename SfinaeExpr, typename Result = void>
using Require = typename _Require<SfinaeExpr, Result>::Result;
 
template <int priority>
struct Priority : Priority<priority - 1>
{
};
 
template <>
struct Priority<0>
{
};
 
template <typename T>
const char* TypeName()
{
    // DisplayName string looks like
    // "interfaces/capnp/common.capnp:ChainNotifications.resendWalletTransactions$Results"
    // This discards the part of the string before the first ':' character.
    // Another alternative would be to use the displayNamePrefixLength field,
    // but this discards everything before the last '.' character, throwing away
    // the object name, which is useful.
    const char* display_name = ::capnp::Schema::from<T>().getProto().getDisplayName().cStr();
    const char* short_name = strchr(display_name, ':');
    return short_name ? short_name + 1 : display_name;
}
 
template <typename T>
struct PtrOrValue {
    std::variant<T*, T> data;
 
    template <typename... Args>
    PtrOrValue(T* ptr, Args&&... args) : data(std::in_place_type<T*>, ptr)
    {
        if (!ptr) data.template emplace<T>(std::forward<Args>(args)...);
    }
 
    T& operator*() { return data.index() ? std::get<T>(data) : *std::get<T*>(data); }
    T* operator->() { return &**this; }
    T& operator*() const { return data.index() ? std::get<T>(data) : *std::get<T*>(data); }
    T* operator->() const { return &**this; }
};
 
// Annotated mutex and lock class (https://clang.llvm.org/docs/ThreadSafetyAnalysis.html)
#if defined(__clang__) && (!defined(SWIG))
#define MP_TSA(x)   __attribute__((x))
#else
#define MP_TSA(x)   // no-op
#endif
 
#define MP_CAPABILITY(x)        MP_TSA(capability(x))
#define MP_SCOPED_CAPABILITY    MP_TSA(scoped_lockable)
#define MP_REQUIRES(x)          MP_TSA(requires_capability(x))
#define MP_ACQUIRE(...)         MP_TSA(acquire_capability(__VA_ARGS__))
#define MP_RELEASE(...)         MP_TSA(release_capability(__VA_ARGS__))
#define MP_ASSERT_CAPABILITY(x) MP_TSA(assert_capability(x))
#define MP_GUARDED_BY(x)        MP_TSA(guarded_by(x))
#define MP_NO_TSA               MP_TSA(no_thread_safety_analysis)
 
class MP_CAPABILITY("mutex") Mutex {
public:
    void lock() MP_ACQUIRE() { m_mutex.lock(); }
    void unlock() MP_RELEASE() { m_mutex.unlock(); }
 
    std::mutex m_mutex;
};
 
class MP_SCOPED_CAPABILITY Lock {
public:
    explicit Lock(Mutex& m) MP_ACQUIRE(m) : m_lock(m.m_mutex) {}
    ~Lock() MP_RELEASE() = default;
    void unlock() MP_RELEASE() { m_lock.unlock(); }
    void lock() MP_ACQUIRE() { m_lock.lock(); }
    void assert_locked(Mutex& mutex) MP_ASSERT_CAPABILITY() MP_ASSERT_CAPABILITY(mutex)
    {
        assert(m_lock.mutex() == &mutex.m_mutex);
        assert(m_lock);
    }
 
    std::unique_lock<std::mutex> m_lock;
};
 
template<typename T>
struct GuardedRef
{
    Mutex& mutex;
    T& ref MP_GUARDED_BY(mutex);
};
 
// CTAD for Clang 16: GuardedRef{mutex, x} -> GuardedRef<decltype(x)>
template <class U>
GuardedRef(Mutex&, U&) -> GuardedRef<U>;
 
template <typename Lock>
struct UnlockGuard
{
    UnlockGuard(Lock& lock) : m_lock(lock) { m_lock.unlock(); }
    ~UnlockGuard() { m_lock.lock(); }
    Lock& m_lock;
};
 
template <typename Lock, typename Callback>
void Unlock(Lock& lock, Callback&& callback)
{
    const UnlockGuard<Lock> unlock(lock);
    callback();
}
 
template <typename Fn, typename After>
decltype(auto) TryFinally(Fn&& fn, After&& after)
{
    bool success{false};
    using R = std::invoke_result_t<Fn>;
    try {
        if constexpr (std::is_void_v<R>) {
            std::forward<Fn>(fn)();
            success = true;
            std::forward<After>(after)();
            return;
        } else {
            decltype(auto) result = std::forward<Fn>(fn)();
            success = true;
            std::forward<After>(after)();
            return result;
        }
    } catch (...) {
        if (!success) std::forward<After>(after)();
        throw;
    }
}
 
void SetOsThreadName(const char* name);
 
std::string ThreadName(const char* exe_name);
 
std::string LogEscape(const kj::StringTree& string, size_t max_size);
 
using Stream = kj::Own<kj::AsyncIoStream>;
 
using ProcessId = int;
using SocketId = int;
inline constexpr SocketId SocketError{-1};
 
using SpawnConnectInfo = std::string;
 
using SpawnConnectInfoToArgsFn = std::function<std::vector<std::string>(const SpawnConnectInfo&)>;
 
std::tuple<ProcessId, SocketId> SpawnProcess(SpawnConnectInfoToArgsFn&& connect_info_to_args);
 
SocketId StartSpawned(const SpawnConnectInfo& connect_info);
 
std::array<SocketId, 2> SocketPair();
 
ProcessId StartProcess(const std::vector<std::string>& args);
 
int WaitProcess(ProcessId pid);
 
inline char* CharCast(char* c) { return c; }
inline char* CharCast(unsigned char* c) { return (char*)c; }
inline const char* CharCast(const char* c) { return c; }
inline const char* CharCast(const unsigned char* c) { return (const char*)c; }
 
#if __has_include(<cxxabi.h>)   // GCC & Clang ─ use <cxxabi.h> to demangle
inline std::string _demangle(const char* m)
{
    int status = 0;
    std::unique_ptr<char, void(*)(void*)> p{
        abi::__cxa_demangle(m, /*output_buffer=*/nullptr, /*length=*/nullptr, &status), std::free};
    return (status == 0 && p) ? p.get() : m;   // fall back on mangled if needed
}
#else                           // MSVC or other ─ no demangling available
inline std::string _demangle(const char* m) { return m; }
#endif
 
template<class T>
std::string CxxTypeName(const T& /*unused*/)
{
#ifdef __cpp_rtti
    return _demangle(typeid(std::decay_t<T>).name());
#else
    return "<type information unavailable without rtti>";
#endif
}
 
struct InterruptException final : std::exception {
    explicit InterruptException(std::string message) : m_message(std::move(message)) {}
    const char* what() const noexcept override { return m_message.c_str(); }
    std::string m_message;
};
 
class CancelProbe;
 
class CancelMonitor
{
public:
    inline ~CancelMonitor();
    inline void promiseDestroyed(CancelProbe& probe);
 
    bool m_canceled{false};
    std::function<void()> m_on_cancel;
    CancelProbe* m_probe{nullptr};
};
 
class CancelProbe
{
public:
    CancelProbe(CancelMonitor& monitor) : m_monitor(&monitor)
    {
        assert(!monitor.m_probe);
        monitor.m_probe = this;
    }
    ~CancelProbe()
    {
        if (m_monitor) m_monitor->promiseDestroyed(*this);
    }
    CancelMonitor* m_monitor;
};
 
CancelMonitor::~CancelMonitor()
{
    if (m_probe) {
        assert(m_probe->m_monitor == this);
        m_probe->m_monitor = nullptr;
        m_probe = nullptr;
    }
}
 
void CancelMonitor::promiseDestroyed(CancelProbe& probe)
{
    // If promise is being destroyed, assume the promise has been canceled. In
    // theory this method could be called when a promise was fulfilled or
    // rejected rather than canceled, but it's safe to assume that's not the
    // case because the CancelMonitor class is meant to be used inside code
    // fulfilling or rejecting the promise and destroyed before doing so.
    assert(m_probe == &probe);
    m_canceled = true;
    if (m_on_cancel) m_on_cancel();
    m_probe = nullptr;
}
} // namespace mp
 
#endif // MP_UTIL_H