From 5de62bbe4dddc64895ddb17c4eb3572c3c9be248 Mon Sep 17 00:00:00 2001
From: Yong He <yonghe@outlook.com>
Date: Mon, 19 Feb 2018 19:53:45 -0500
Subject: more to fixing memory leaks 1. reorder destruction order of several
 key classes to avoid using deleted IR objects when destroying Types 2. remove
 Session::canonicalTypes and make each Type own a RefPtr to the canonicalType,
 to allow types to be destroyed along with each IRModule it belongs to.

---
 source/core/smart-pointer.h | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'source/core/smart-pointer.h')

diff --git a/source/core/smart-pointer.h b/source/core/smart-pointer.h
index 17d6caaa4..bc1683a5b 100644
--- a/source/core/smart-pointer.h
+++ b/source/core/smart-pointer.h
@@ -35,6 +35,11 @@ namespace Slang
             referenceCount++;
         }
 
+        void decreaseReference()
+        {
+            --referenceCount;
+        }
+
         void releaseReference()
         {
             SLANG_ASSERT(referenceCount != 0);
@@ -192,6 +197,15 @@ namespace Slang
             return pointer;
         }
 
+        T* detach()
+        {
+            if (pointer)
+                dynamic_cast<RefObject*>(pointer)->decreaseReference();
+            auto rs = pointer;
+            pointer = nullptr;
+            return rs;
+        }
+
     private:
         T* pointer;
         
-- 
cgit v1.2.3