summaryrefslogtreecommitdiff
path: root/core/fxcrt
diff options
context:
space:
mode:
Diffstat (limited to 'core/fxcrt')
-rw-r--r--core/fxcrt/cfx_unowned_ptr.h18
-rw-r--r--core/fxcrt/cfx_unowned_ptr_unittest.cpp32
2 files changed, 43 insertions, 7 deletions
diff --git a/core/fxcrt/cfx_unowned_ptr.h b/core/fxcrt/cfx_unowned_ptr.h
index 11f4e8e38b..bc58144335 100644
--- a/core/fxcrt/cfx_unowned_ptr.h
+++ b/core/fxcrt/cfx_unowned_ptr.h
@@ -26,17 +26,20 @@ class CFX_UnownedPtr {
CFX_UnownedPtr(std::nullptr_t ptr) {}
#if defined(MEMORY_TOOL_REPLACES_ALLOCATOR)
- ~CFX_UnownedPtr() {
- if (m_pObj)
- reinterpret_cast<volatile uint8_t*>(m_pObj)[0];
- }
+ ~CFX_UnownedPtr() { Probe(); }
#endif
CFX_UnownedPtr& operator=(T* that) {
+#if defined(MEMORY_TOOL_REPLACES_ALLOCATOR)
+ Probe();
+#endif
m_pObj = that;
return *this;
}
CFX_UnownedPtr& operator=(const CFX_UnownedPtr& that) {
+#if defined(MEMORY_TOOL_REPLACES_ALLOCATOR)
+ Probe();
+#endif
if (*this != that)
m_pObj = that.Get();
return *this;
@@ -58,6 +61,13 @@ class CFX_UnownedPtr {
T* operator->() const { return m_pObj; }
private:
+#if defined(MEMORY_TOOL_REPLACES_ALLOCATOR)
+ void Probe() {
+ if (m_pObj)
+ reinterpret_cast<volatile uint8_t*>(m_pObj)[0];
+ }
+#endif
+
T* m_pObj = nullptr;
};
diff --git a/core/fxcrt/cfx_unowned_ptr_unittest.cpp b/core/fxcrt/cfx_unowned_ptr_unittest.cpp
index 7d29faedf5..586c4bbb2e 100644
--- a/core/fxcrt/cfx_unowned_ptr_unittest.cpp
+++ b/core/fxcrt/cfx_unowned_ptr_unittest.cpp
@@ -25,6 +25,15 @@ void DeleteDangling() {
delete ptr2;
}
+void AssignDangling() {
+ Clink* ptr1 = new Clink();
+ Clink* ptr2 = new Clink();
+ ptr2->next_ = ptr1;
+ delete ptr1;
+ ptr2->next_ = nullptr;
+ delete ptr2;
+}
+
} // namespace
TEST(fxcrt, UnownedPtrOk) {
@@ -43,7 +52,24 @@ TEST(fxcrt, UnownedPtrNotOk) {
#endif
}
-TEST(fxcrt, OperatorEQ) {
+TEST(fxcrt, UnownedAssignOk) {
+ Clink* ptr1 = new Clink();
+ Clink* ptr2 = new Clink();
+ ptr2->next_ = ptr1;
+ ptr2->next_ = nullptr;
+ delete ptr2;
+ delete ptr1;
+}
+
+TEST(fxcrt, UnownedAssignNotOk) {
+#if defined(MEMORY_TOOL_REPLACES_ALLOCATOR)
+ EXPECT_DEATH(AssignDangling(), "");
+#else
+ AssignDangling();
+#endif
+}
+
+TEST(fxcrt, UnownedOperatorEQ) {
int foo;
CFX_UnownedPtr<int> ptr1;
EXPECT_TRUE(ptr1 == ptr1);
@@ -59,7 +85,7 @@ TEST(fxcrt, OperatorEQ) {
EXPECT_TRUE(ptr1 == ptr3);
}
-TEST(fxcrt, OperatorNE) {
+TEST(fxcrt, UnownedOperatorNE) {
int foo;
CFX_UnownedPtr<int> ptr1;
EXPECT_FALSE(ptr1 != ptr1);
@@ -75,7 +101,7 @@ TEST(fxcrt, OperatorNE) {
EXPECT_FALSE(ptr1 != ptr3);
}
-TEST(fxcrt, OperatorLT) {
+TEST(fxcrt, UnownedOperatorLT) {
int foos[2];
CFX_UnownedPtr<int> ptr1(&foos[0]);
CFX_UnownedPtr<int> ptr2(&foos[1]);