1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106
|
// RUN: %target-build-swift %s %import-libdispatch -o %t_binary
// RUN: %target-codesign %t_binary
// RUN: %target-run %t_binary
// REQUIRES: executable_test
// REQUIRES: stress_test
// REQUIRES: libdispatch
// UNSUPPORTED: threading_none
import StdlibUnittest
import Dispatch
let iterations = 200_000
class Thing {}
class WBox<T: AnyObject> {
weak var wref: T?
init(_ ref: T) { self.wref = ref }
init() { self.wref = nil }
}
let WeakReferenceRaceTests = TestSuite("WeakReferenceRaceTests")
func forwardOptional<T>(_ t: T!) -> T {
return t!
}
WeakReferenceRaceTests.test("class instance property (copy)") {
let q = DispatchQueue(label: "", attributes: .concurrent)
// Capture a weak reference via its container object.
// https://github.com/apple/swift/issues/42814
for i in 1...iterations {
let box = WBox(Thing())
let closure = {
let nbox = WBox<Thing>()
nbox.wref = box.wref
_blackHole(nbox)
}
q.async(execute: closure)
q.async(execute: closure)
}
q.async(flags: .barrier) {}
}
WeakReferenceRaceTests.test("class instance property (load)") {
let q = DispatchQueue(label: "", attributes: .concurrent)
// Capture a weak reference via its container object.
// https://github.com/apple/swift/issues/42814
for i in 1...iterations {
let box = WBox(Thing())
let closure = {
if let ref = box.wref {
_blackHole(ref)
}
}
q.async(execute: closure)
q.async(execute: closure)
}
q.async(flags: .barrier) {}
}
WeakReferenceRaceTests.test("direct capture (copy)") {
let q = DispatchQueue(label: "", attributes: .concurrent)
// Capture a weak reference directly in multiple closures
for i in 1...iterations {
weak var wref = Thing()
let closure = {
let nbox = WBox<Thing>()
nbox.wref = wref
_blackHole(nbox)
}
q.async(execute: closure)
q.async(execute: closure)
}
q.async(flags: .barrier) {}
}
WeakReferenceRaceTests.test("direct capture (load)") {
let q = DispatchQueue(label: "", attributes: .concurrent)
// Capture a weak reference directly in multiple closures
for i in 1...iterations {
weak var wref = Thing()
let closure = {
if let ref = wref {
_blackHole(ref)
}
}
q.async(execute: closure)
q.async(execute: closure)
}
q.async(flags: .barrier) {}
}
runAllTests()
|