Merge pull request #188 from p-x9/feature/dyld-cache-tpro-mapping

Author: p-x9

Sources/MachOKit/DyldCache.swift

@@ -390,6 +390,17 @@ extension DyldCache {
         }
         return fileHandle.read(offset: offset)
     }
+
+    public var tproMappings: DataSequence<DyldCacheTproMappingInfo>? {
+        guard mainCacheHeader.tproMappingsOffset > 0,
+              mainCacheHeader.hasProperty(\.tproMappingsCount) else {
+            return nil
+        }
+        return fileHandle.readDataSequence(
+            offset: numericCast(header.tproMappingsOffset),
+            numberOfElements: numericCast(header.tproMappingsCount)
+        )
+    }
 }
 
 extension DyldCache {

Sources/MachOKit/DyldCacheLoaded.swift

@@ -392,4 +392,17 @@ extension DyldCacheLoaded {
             .advanced(by: numericCast(header.swiftOptsOffset))
             .autoBoundPointee()
     }
+
+    public var tproMappings: MemorySequence<DyldCacheTproMappingInfo>? {
+        guard mainCacheHeader.tproMappingsOffset > 0,
+              mainCacheHeader.hasProperty(\.tproMappingsCount) else {
+            return nil
+        }
+        return .init(
+            basePointer: ptr
+                .advanced(by: numericCast(header.tproMappingsOffset))
+                .assumingMemoryBound(to: DyldCacheTproMappingInfo.self),
+            numberOfElements: numericCast(header.tproMappingsCount)
+        )
+    }
 }

Sources/MachOKit/Model/DyldCache/DyldCacheTproMappingInfo.swift

@@ -0,0 +1,15 @@
+//
+//  DyldCacheTproMappingInfo.swift
+//  MachOKit
+//
+//  Created by p-x9 on 2025/02/27
+//  
+//
+
+import Foundation
+
+public struct DyldCacheTproMappingInfo: LayoutWrapper {
+    public typealias Layout = dyld_cache_tpro_mapping_info
+
+    public var layout: Layout
+}

Sources/MachOKit/Protocol/DyldCacheRepresentable.swift

@@ -16,6 +16,7 @@ public protocol DyldCacheRepresentable {
     associatedtype SubCaches: RandomAccessCollection<DyldSubCacheEntry>
     associatedtype DylibsTrie: TrieTreeProtocol<DylibsTrieNodeContent>
     associatedtype ProgramsTrie: TrieTreeProtocol<ProgramsTrieNodeContent>
+    associatedtype TproMappingInfos: RandomAccessCollection<DyldCacheTproMappingInfo>
 
     /// Byte size of header
     var headerSize: Int { get }
@@ -97,6 +98,9 @@ public protocol DyldCacheRepresentable {
     /// [dyld implementation](https://github.com/apple-oss-distributions/dyld/blob/65bbeed63cec73f313b1d636e63f243964725a9d/common/DyldSharedCache.cpp#L2088-L2098)
     var swiftOptimization: SwiftOptimization? { get }
 
+    /// Sequence of tpro mapping infos
+    var tproMappings: TproMappingInfos? { get }
+
     /// Get the prebuiltLoaderSet indicated by programOffset.
     /// - Parameter programOffset: program name and offset pair
     /// - Returns: prebuiltLoaderSet

Tests/MachOKitTests/DyldCacheLoadedPrintTests.swift

@@ -318,6 +318,13 @@ final class DyldCacheLoadedPrintTests: XCTestCase {
         print("Metadata Conformance Hash Table Cache Offset:", swiftOptimization.metadataConformanceHashTableCacheOffset)
         print("Foreign Type Conformance Hash Table Cache Offset:", swiftOptimization.foreignTypeConformanceHashTableCacheOffset)
     }
+
+    func testTproMappings() throws {
+        guard let mappings = cache.tproMappings else { return }
+        for mapping in mappings {
+            print("- 0x\(String(mapping.unslidAddress, radix: 16)), \(mapping.size)")
+        }
+    }
 }
 
 #endif

Tests/MachOKitTests/DyldCachePrintTests.swift

@@ -298,6 +298,13 @@ final class DyldCachePrintTests: XCTestCase {
         print("Metadata Conformance Hash Table Cache Offset:", swiftOptimization.metadataConformanceHashTableCacheOffset)
         print("Foreign Type Conformance Hash Table Cache Offset:", swiftOptimization.foreignTypeConformanceHashTableCacheOffset)
     }
+
+    func testTproMappings() throws {
+        guard let mappings = cache.tproMappings else { return }
+        for mapping in mappings {
+            print("- 0x\(String(mapping.unslidAddress, radix: 16)), \(mapping.size)")
+        }
+    }
 }
 
 extension DyldCachePrintTests {