Avoid potential null dereference in annotated_ptr

Fixes [BUG]: UB in annotated_ptr #2942
NVIDIA · Nov 25, 2024 · 7d4d5cf · 7d4d5cf
1 parent 80031e2
commit 7d4d5cf
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/libcudacxx/include/cuda/annotated_ptr b/libcudacxx/include/cuda/annotated_ptr
@@ -391,6 +391,10 @@ public:
 
   _CCCL_HOST_DEVICE pointer get() const noexcept
   {
+    if (__repr == nullptr)
+    {
+      return nullptr;
+    }
     constexpr bool __is_shared = std::is_same<_Property, access_property::shared>::value;
     return __is_shared ? __repr : &(*annotated_ptr<value_type, access_property::global>(__repr));
   }