Merge pull request #154 from p-x9/feature/image-path-objc-ro-header

Fix to get mach-o image name from `objc_header_info_ro_t`

Author: p-x9

Sources/MachOKit/Model/DyldCache/ObjCOptimization/ObjCHeaderInfoRO.swift

@@ -97,9 +97,10 @@ public struct ObjCHeaderInfoRO64: LayoutWrapper, ObjCHeaderInfoROProtocol {
         guard let offset = resolvedMachOHeaderOffset(in: cache) else {
             return nil
         }
+        let imagePath = imagePath(in: cache)
         return try? .init(
             url: cache.url,
-            imagePath: "", // FIXME: path
+            imagePath: imagePath ?? "",
             headerStartOffsetInCache: numericCast(offset)
         )
     }
@@ -168,9 +169,10 @@ public struct ObjCHeaderInfoRO32: LayoutWrapper, ObjCHeaderInfoROProtocol {
         guard let offset = resolvedMachOHeaderOffset(in: cache) else {
             return nil
         }
+        let imagePath = imagePath(in: cache)
         return try? .init(
             url: cache.url,
-            imagePath: "", // FIXME: path
+            imagePath: imagePath ?? "",
             headerStartOffsetInCache: numericCast(offset)
         )
     }
@@ -199,4 +201,18 @@ extension ObjCHeaderInfoROProtocol {
         }
         return fileOffset
     }
+
+    internal func imagePath(in cache: DyldCache) -> String? {
+        let offset = offset + machOHeaderOffset
+        let address = cache.mainCacheHeader.sharedRegionStart + numericCast(offset)
+        guard let imageInfos = cache.imageInfos,
+              let imageInfo = imageInfos.first(
+                where: {
+                    $0.address == address
+                }
+              ) else {
+            return nil
+        }
+        return imageInfo.path(in: cache)
+    }
 }

Tests/MachOKitTests/DyldCachePrintTests.swift

@@ -285,11 +285,7 @@ final class DyldCachePrintTests: XCTestCase {
             XCTAssertEqual(info.mhdr_offset, _info.mhdr_offset)
             XCTAssertEqual(info.info_offset, _info.info_offset)
 
-            let path = machO.loadCommands
-                .info(of: LoadCommand.idDylib)?
-                .dylib(in: machO)
-                .name ?? "unknonw"
-            print(" \(path), offset: \(machO.headerStartOffsetInCache)")
+            print(" \(machO.imagePath), offset: \(machO.headerStartOffsetInCache)")
         }
     }