TSharedPtr::As()&: leakage fix

По следам https://a.yandex-team.ru/review/2996409/details
Исправлены случаи при неудачных кастах:
1. контрольный блок дёргается даже при указании на nullptr (нехорошо для атомиков)
2. если исходный шаред поинтер разрушится раньше, чем полученный nullptr (например, при возврате из функции), то ресурс утечёт

1 files changed, 15 insertions, 6 deletions

diff --git a/util/generic/ptr.h b/util/generic/ptr.h
index b35d54d780..cc2e3c0f51 100644
--- a/util/generic/ptr.h
+++ b/util/generic/ptr.h
@@ -900,18 +900,27 @@ public:
     [[nodiscard]] inline TSharedPtr<TT, C, D> As() & noexcept {
         static_assert(std::has_virtual_destructor<TT>(), "Type should have a virtual dtor");
         static_assert(std::is_base_of<T, TT>(), "When downcasting from T to TT, T should be a parent of TT");
-        Ref();
-        return TSharedPtr<TT, C, D>(dynamic_cast<TT*>(T_), C_);
+        if (const auto ttPtr = dynamic_cast<TT*>(T_)) {
+            TSharedPtr<TT, C, D> ttSharedPtr(ttPtr, C_);
+            ttSharedPtr.Ref();
+            return ttSharedPtr;
+        } else {
+            return TSharedPtr<TT, C, D>{};
+        }
     }
 
     template <class TT>
     [[nodiscard]] inline TSharedPtr<TT, C, D> As() && noexcept {
         static_assert(std::has_virtual_destructor<TT>(), "Type should have a virtual dtor");
         static_assert(std::is_base_of<T, TT>(), "When downcasting from T to TT, T should be a parent of TT");
-        auto resultPtr = TSharedPtr<TT, C, D>(dynamic_cast<TT*>(T_), C_);
-        T_ = nullptr;
-        C_ = nullptr;
-        return resultPtr;
+        if (const auto ttPtr = dynamic_cast<TT*>(T_)) {
+            TSharedPtr<TT, C, D> ttSharedPtr(ttPtr, C_);
+            T_ = nullptr;
+            C_ = nullptr;
+            return ttSharedPtr;
+        } else {
+            return TSharedPtr<TT, C, D>{};
+        }
     }
 
 #ifdef __cpp_impl_three_way_comparison