averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-14 20:36:38 +01:00
Code Issues Packages Projects Releases Wiki Activity

[embedded] Implement retain_n and release_n in the embedded runtime

Browse Source
This commit is contained in:
Kuba Mracek
2023-09-27 22:55:43 -07:00
parent 1470023add
commit bcd7d3f9ec
1 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
stdlib/public/core/EmbeddedRuntime.swift
View File

@@ -101,21 +101,31 @@ public func swift_isUniquelyReferenced_nonNull_native(object: UnsafeMutablePoint
@_silgen_name("swift_retain")
public func swift_retain(object: Builtin.RawPointer) -> Builtin.RawPointer {
return swift_retain_n(object: object, n: 1)
}
@_silgen_name("swift_retain_n")
public func swift_retain_n(object: Builtin.RawPointer, n: UInt32) -> Builtin.RawPointer {
if Int(Builtin.ptrtoint_Word(object)) == 0 { return object }
let o = UnsafeMutablePointer<HeapObject>(object)
// TODO/FIXME: Refcounting is not thread-safe, the following only works in single-threaded environments.
if o.pointee.refcount == HeapObject.immortalRefCount { return o._rawValue }
o.pointee.refcount += 1
o.pointee.refcount += Int(n)
return o._rawValue
}
@_silgen_name("swift_release")
public func swift_release(object: Builtin.RawPointer) {
swift_release_n(object: object, n: 1)
}
@_silgen_name("swift_release_n")
public func swift_release_n(object: Builtin.RawPointer, n: UInt32) {
if Int(Builtin.ptrtoint_Word(object)) == 0 { return }
let o = UnsafeMutablePointer<HeapObject>(object)
// TODO/FIXME: Refcounting is not thread-safe, the following only works in single-threaded environments.
if o.pointee.refcount == HeapObject.immortalRefCount { return }
o.pointee.refcount -= 1
o.pointee.refcount -= Int(n)
if (o.pointee.refcount & HeapObject.refcountMask) == 0 {
_swift_embedded_invoke_heap_object_destroy(o)
}