fix: wrongly dispose unowned translation unit

Author: ddddxxx

Sources/Clang/Cursor.swift

@@ -197,7 +197,7 @@ extension Cursor {
 
   /// Returns the translation unit that a cursor originated from.
   public var translationUnit: TranslationUnit {
-    return TranslationUnit(clang: clang_Cursor_getTranslationUnit(asClang()))
+    return TranslationUnit(clang: clang_Cursor_getTranslationUnit(asClang()), owned: false)
   }
 
   /// Retrieves all the children of the provided cursor.

Sources/Clang/TranslationUnit.swift

@@ -126,9 +126,11 @@ public struct TranslationUnitSaveOptions: OptionSet {
 
 public class TranslationUnit {
   let clang: CXTranslationUnit
+  private let owned: Bool
 
-  init(clang: CXTranslationUnit) {
+  init(clang: CXTranslationUnit, owned: Bool) {
     self.clang = clang
+    self.owned = owned
   }
 
 
@@ -162,6 +164,7 @@ public class TranslationUnit {
       }
       return unit!
     }
+    self.owned = true
   }
 
   /// Creates a `TranslationUnit` by parsing the source code passed,
@@ -236,6 +239,7 @@ public class TranslationUnit {
     }
 
     self.clang = unit!
+    self.owned = true
   }
 
   /// Retrieve the cursor that represents the given translation unit.
@@ -471,6 +475,8 @@ public class TranslationUnit {
   }
 
   deinit {
-    clang_disposeTranslationUnit(clang)
+    if owned {
+      clang_disposeTranslationUnit(clang)
+    }
   }
 }

Tests/ClangTests/ClangTests.swift

@@ -235,6 +235,19 @@ class ClangTests: XCTestCase {
             XCTFail("\(error)")
         }
     }
+  
+  func testDisposeTranslateUnit() {
+    do {
+      let filename = "input_tests/init-ast.c"
+      let unit = try TranslationUnit(filename: filename)
+      let cursor = unit.cursor
+      for _ in 0..<2 {
+        _ = cursor.translationUnit
+      }
+    } catch {
+      XCTFail("\(error)")
+    }
+  }
 
   static var allTests : [(String, (ClangTests) -> () throws -> Void)] {
     return [