Add a utility function to Inspector to load the reflection metadata from

loaded images

Author: aschwaighofer

tools/swift-inspect/Sources/swift-inspect/Inspector.swift

@@ -34,6 +34,15 @@ class Inspector {
     mach_port_deallocate(mach_task_self_, task)
   }
 
+  func addReflectionInfoFromLoadedImages(context: SwiftReflectionContextRef) {
+    var numSuccessfulImages = 0
+    CSSymbolicatorForeachSymbolOwnerAtTime(symbolicator, kCSNow, { owner in
+      let address = CSSymbolOwnerGetBaseAddress(owner);
+      let name = CSSymbolOwnerGetName(owner)
+      let _ = swift_reflection_addImage(context, address)
+      })
+  }
+
   static func findTask(_ pid: pid_t, tryForkCorpse: Bool) -> task_t {
     var task = task_t()
     var kr = task_for_pid(mach_task_self_, pid, &task)

tools/swift-inspect/Sources/swift-inspect/symbolication.swift

@@ -69,6 +69,13 @@ enum Sym {
     symbol(symbolicationHandle, "task_peek_string")
   static let task_stop_peeking: @convention(c) (task_t) -> kern_return_t =
     symbol(symbolicationHandle, "task_stop_peeking")
+
+  static let CSSymbolicatorForeachSymbolOwnerAtTime:
+    @convention(c) (CSSymbolicatorRef, CSMachineTime, @convention(block) (CSSymbolOwnerRef) -> Void) -> UInt =
+      symbol(coreSymbolicationHandle, "CSSymbolicatorForeachSymbolOwnerAtTime")
+
+  static let CSSymbolOwnerGetBaseAddress: @convention(c) (CSSymbolOwnerRef) -> mach_vm_address_t =
+    symbol(symbolicationHandle, "CSSymbolOwnerGetBaseAddress")
 }
 
 typealias CSMachineTime = UInt64
@@ -145,6 +152,19 @@ func CSSymbolicatorGetSymbolWithAddressAtTime(
   Sym.CSSymbolicatorGetSymbolWithAddressAtTime(symbolicator, address, time)
 }
 
+func CSSymbolicatorForeachSymbolOwnerAtTime(
+  _ symbolicator: CSSymbolicatorRef,
+  _ time: CSMachineTime,
+  _ symbolIterator: (CSSymbolOwnerRef) -> Void
+  ) ->  UInt {
+      return Sym.CSSymbolicatorForeachSymbolOwnerAtTime(symbolicator, time,
+                                                        symbolIterator)
+}
+
+func CSSymbolOwnerGetBaseAddress(_ symbolOwner: CSSymbolOwnerRef) -> mach_vm_address_t {
+    return Sym.CSSymbolOwnerGetBaseAddress(symbolOwner)
+}
+
 func task_start_peeking(_ task: task_t) -> Bool {
   let result = Sym.task_start_peeking(task)
   if result == KERN_SUCCESS {