Merge pull request #40460 from nate-chandler/lexical_lifetimes/test/verification/20211207/1

[Test] Add two lexical lifetime verification tests.

Author: nate-chandler

validation-test/SILOptimizer/lexical-lifetimes.swift

@@ -21,6 +21,64 @@ class D {
   }
 }
 
+class DataWrapper {
+
+    var pointer: UnsafeMutableRawBufferPointer
+
+    init(count: Int) {
+        pointer = UnsafeMutableRawBufferPointer.allocate(byteCount: count, alignment: MemoryLayout<Int>.alignment)
+    }
+
+    var bytes: UnsafeMutableRawBufferPointer { return pointer }
+
+    deinit {
+        pointer.deallocate()
+    }
+}
+
+var fileHandleMap: [Int32 : String] = [:]
+
+func openTheFile(_ path: String) -> Int32 {
+  let fd: Int32 = 42
+  assert(fileHandleMap[fd] == nil)
+  fileHandleMap[fd] = path
+  return fd
+}
+
+func closeTheFile(_ fd: Int32) {
+  fileHandleMap.removeValue(forKey: fd)
+}
+
+func writeToTheFile(_ fd: Int32) {
+  assert(fileHandleMap[fd] != nil)
+}
+
+class FileHandleWrapper {
+
+    var handle: Int32? = nil
+
+    func open(path: String) {
+        let fd = openTheFile(path)
+        if fd >= 0 {
+            handle = fd
+        }
+    }
+
+    func close() {
+        if let fd = handle {
+            closeTheFile(fd)
+            handle = nil
+        }
+    }
+
+    deinit {
+        if let fd = handle {
+            closeTheFile(fd)
+        }
+    }
+}
+
+
 // =============================================================================
 // = Declarations                                                           }} =
 // =============================================================================
@@ -29,27 +87,71 @@ class D {
 // = Tests                                                                  {{ =
 // =============================================================================
 
-func test_localLetKeepsObjectAliveBeyondCallToClassWithWeakReference() {
+func test_localLet_keepsObjectAliveBeyondCallToClassWithWeakReference() {
   let d = D()
   let c = C(d)
-  // CHECK: cWillFoo{{.*}} test_localLetKeepsObjectAliveBeyondCallToClassWithWeakReference
+  // CHECK: cWillFoo{{.*}} test_localLet_keepsObjectAliveBeyondCallToClassWithWeakReference
   c.foo(#function)
 }
 
-func test_localVarKeepsObjectAliveBeyondCallToClassWithWeakReference() {
+func test_localVar_keepsObjectAliveBeyondCallToClassWithWeakReference() {
   var d = D()
   let c = C(d)
-  // CHECK: cWillFoo{{.*}} test_localVarKeepsObjectAliveBeyondCallToClassWithWeakReference
+  // CHECK: cWillFoo{{.*}} test_localVar_keepsObjectAliveBeyondCallToClassWithWeakReference
   c.foo(#function)
 }
 
+func test_localLet_keepsObjectAliveBeyondCallToClassWithPointer_doit(_ input: [UInt8]) {
+    let data = DataWrapper(count: input.count)
+    data.bytes.copyBytes(from: input)
+}
+func test_localLet_keepsObjectAliveBeyondCallToClassWithPointer() {
+  test_localLet_keepsObjectAliveBeyondCallToClassWithPointer_doit([1,2,3,4,50])
+}
+
+func test_localVar_keepsObjectAliveBeyondCallToClassWithPointer_doit(_ input: [UInt8]) {
+    let data = DataWrapper(count: input.count)
+    data.bytes.copyBytes(from: input)
+}
+func test_localVar_keepsObjectAliveBeyondCallToClassWithPointer() {
+  test_localVar_keepsObjectAliveBeyondCallToClassWithPointer_doit([1,2,3,4,50])
+}
+
+func test_localLet_keepsObjectAliveBeyondCallToSynchronizationPointFunction_doit(_ path: String) {
+    var file = FileHandleWrapper()
+    file.open(path: path)
+    // Retrieving 'fd' is the last use of 'file'
+    guard let fd = file.handle else { return }
+    // 'fd' has now been closed. The subsequent write will fail.
+    writeToTheFile(fd)
+}
+func test_localLet_keepsObjectAliveBeyondCallToSynchronizationPointFunction() {
+  test_localLet_keepsObjectAliveBeyondCallToSynchronizationPointFunction_doit("blue")
+}
+
+func test_localVar_keepsObjectAliveBeyondCallToSynchronizationPointFunction_doit(_ path: String) {
+    var file = FileHandleWrapper()
+    file.open(path: path)
+    // Retrieving 'fd' is the last use of 'file'
+    guard let fd = file.handle else { return }
+    // 'fd' has now been closed. The subsequent write will fail.
+    writeToTheFile(fd)
+}
+func test_localVar_keepsObjectAliveBeyondCallToSynchronizationPointFunction() {
+  test_localVar_keepsObjectAliveBeyondCallToSynchronizationPointFunction_doit("blue")
+}
+
 // =============================================================================
 // = Tests                                                                  }} =
 // =============================================================================
 
 func run() {
-  test_localLetKeepsObjectAliveBeyondCallToClassWithWeakReference()
-  test_localVarKeepsObjectAliveBeyondCallToClassWithWeakReference()
+  test_localLet_keepsObjectAliveBeyondCallToClassWithWeakReference()
+  test_localVar_keepsObjectAliveBeyondCallToClassWithWeakReference()
+  test_localLet_keepsObjectAliveBeyondCallToClassWithPointer()
+  test_localVar_keepsObjectAliveBeyondCallToClassWithPointer()
+  test_localLet_keepsObjectAliveBeyondCallToSynchronizationPointFunction()
+  test_localVar_keepsObjectAliveBeyondCallToSynchronizationPointFunction()
 }
 
 run()