Merge pull request #74332 from eeckstein/swift_dynamiccastclass

embedded: add swift_dynamicCastClass runtime function

Author: eeckstein

stdlib/public/core/EmbeddedRuntime.swift

@@ -187,6 +187,20 @@ func isValidPointerForNativeRetain(object: Builtin.RawPointer) -> Bool {
 public func swift_setDeallocating(object: Builtin.RawPointer) {
 }
 
+@_cdecl("swift_dynamicCastClass")
+public func swift_dynamicCastClass(object: UnsafeMutableRawPointer, targetMetadata: UnsafeRawPointer) -> UnsafeMutableRawPointer? {
+  let sourceObj = object.assumingMemoryBound(to: HeapObject.self)
+  var type = _swift_embedded_get_heap_object_metadata_pointer(sourceObj).assumingMemoryBound(to: ClassMetadata.self)
+  let targetType = targetMetadata.assumingMemoryBound(to: ClassMetadata.self)
+  while type != targetType {
+    guard let superType = type.pointee.superclassMetadata else {
+      return nil
+    }
+    type = UnsafeMutablePointer(superType)
+  }
+  return object
+}
+
 @_cdecl("swift_isUniquelyReferenced_native")
 public func swift_isUniquelyReferenced_native(object: Builtin.RawPointer) -> Bool {
   if !isValidPointerForNativeRetain(object: object) { return false }

test/embedded/classes.swift

@@ -15,9 +15,15 @@ class MyClass {
 }
 
 class MySubClass: MyClass {
+  var x = 27
+
   override init() { print("MySubClass.init") }
   deinit { print("MySubClass.deinit") }
   override func foo() { print("MySubClass.foo") }
+
+  func printX() {
+    print(x)
+  }
 }
 
 class MySubSubClass: MySubClass {
@@ -26,6 +32,17 @@ class MySubSubClass: MySubClass {
   override func foo() { print("MySubSubClass.foo") }
 }
 
+class OtherSubClass: MyClass {}
+
+func testCasting(_ title: StaticString, _ c: MyClass) {
+  print(title, terminator: "")
+  if let s = c as? MySubClass {
+    s.printX()
+  } else {
+    print("-")
+  }
+}
+
 @main
 struct Main {
   static var o: (MyClass?, MyClass?, MyClass?) = (nil, nil, nil)
@@ -69,5 +86,14 @@ struct Main {
     // CHECK: MySubClass.deinit
     // CHECK: MyClass.deinit
     print("")
+
+    // CHECK: base: -
+    testCasting("base: ", MyClass())
+    // CHECK: sub: 27
+    testCasting("sub: ", MySubClass())
+    // CHECK: subsub: 27
+    testCasting("subsub: ", MySubSubClass())
+    // CHECK: other: -
+    testCasting("other: ", OtherSubClass())
   }
 }