diff --git a/Telegram/SourceFiles/base/weak_ptr.h b/Telegram/SourceFiles/base/weak_ptr.h index ddab64a2ec..76dd3de83d 100644 --- a/Telegram/SourceFiles/base/weak_ptr.h +++ b/Telegram/SourceFiles/base/weak_ptr.h @@ -11,14 +11,17 @@ https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL #include namespace base { + +class has_weak_ptr; + namespace details { struct alive_tracker { - explicit alive_tracker(const void *value) : value(value) { + explicit alive_tracker(const has_weak_ptr *value) : value(value) { } std::atomic counter = 1; - std::atomic value; + std::atomic value; }; inline alive_tracker *check_and_increment(alive_tracker *tracker) noexcept { @@ -36,8 +39,6 @@ inline void decrement(alive_tracker *tracker) noexcept { } // namespace details -class has_weak_ptr; - template class weak_ptr; @@ -69,8 +70,7 @@ private: details::alive_tracker *incrementAliveTracker() const { auto current = _alive.load(); if (!current) { - auto alive = std::make_unique( - static_cast(this)); + auto alive = std::make_unique(this); if (_alive.compare_exchange_strong(current, alive.get())) { return alive.release(); } @@ -178,15 +178,11 @@ public: } T *get() const noexcept { + const auto strong = _alive ? _alive->value.load() : nullptr; if constexpr (std::is_const_v) { - return _alive - ? static_cast(_alive->value.load()) - : nullptr; + return static_cast(strong); } else { - return _alive - ? const_cast( - static_cast(_alive->value.load())) - : nullptr; + return const_cast(static_cast(strong)); } } explicit operator bool() const noexcept {