Merge pull request #48 from p-x9/fix/address-range

Fix address range
p-x9 · Feb 9, 2024 · a84ec64 · a84ec64
2 parents 7827966 + 32eaff3
commit a84ec64
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/Sources/MachOKit/MachOImage+static.swift b/Sources/MachOKit/MachOImage+static.swift
@@ -3,7 +3,7 @@
 //
 //
 //  Created by p-x9 on 2024/02/06.
-//  
+//
 //
 
 import Foundation
@@ -99,10 +99,11 @@ extension MachOImage {
 fileprivate extension MachOImage {
     var addressRange: ClosedRange<Int>? {
         guard let slide = vmaddrSlide,
-              let start = segments.first?.startPtr(vmaddrSlide: slide),
-              let end = segments.last?.endPtr(vmaddrSlide: slide) else {
+              let text: any SegmentCommandProtocol = loadCommands.text64 ?? loadCommands.text,
+              let start = text.startPtr(vmaddrSlide: slide) else {
             return nil
         }
+        let end = start.advanced(by: text.fileSize)
         return Int(bitPattern: start) ... Int(bitPattern: end)
     }