| ... | @@ -17,6 +17,7 @@ const aarch64 = @import("../arch/aarch64/bits.zig"); | ... | @@ -17,6 +17,7 @@ const aarch64 = @import("../arch/aarch64/bits.zig"); |
| 17 | const bind = @import("MachO/bind.zig"); | 17 | const bind = @import("MachO/bind.zig"); |
| 18 | const codegen = @import("../codegen.zig"); | 18 | const codegen = @import("../codegen.zig"); |
| 19 | const dead_strip = @import("MachO/dead_strip.zig"); | 19 | const dead_strip = @import("MachO/dead_strip.zig"); |
| | 20 | const fat = @import("MachO/fat.zig"); |
| 20 | const link = @import("../link.zig"); | 21 | const link = @import("../link.zig"); |
| 21 | const llvm_backend = @import("../codegen/llvm.zig"); | 22 | const llvm_backend = @import("../codegen/llvm.zig"); |
| 22 | const target_util = @import("../target.zig"); | 23 | const target_util = @import("../target.zig"); |
| ... | @@ -60,6 +61,32 @@ const SystemLib = struct { | ... | @@ -60,6 +61,32 @@ const SystemLib = struct { |
| 60 | weak: bool = false, | 61 | weak: bool = false, |
| 61 | }; | 62 | }; |
| 62 | | 63 | |
| | 64 | const Section = struct { |
| | 65 | header: macho.section_64, |
| | 66 | segment_index: u8, |
| | 67 | |
| | 68 | // TODO is null here necessary, or can we do away with tracking via section |
| | 69 | // size in incremental context? |
| | 70 | last_atom: ?*Atom = null, |
| | 71 | |
| | 72 | /// A list of atoms that have surplus capacity. This list can have false |
| | 73 | /// positives, as functions grow and shrink over time, only sometimes being added |
| | 74 | /// or removed from the freelist. |
| | 75 | /// |
| | 76 | /// An atom has surplus capacity when its overcapacity value is greater than |
| | 77 | /// padToIdeal(minimum_atom_size). That is, when it has so |
| | 78 | /// much extra capacity, that we could fit a small new symbol in it, itself with |
| | 79 | /// ideal_capacity or more. |
| | 80 | /// |
| | 81 | /// Ideal capacity is defined by size + (size / ideal_factor). |
| | 82 | /// |
| | 83 | /// Overcapacity is measured by actual_capacity - ideal_capacity. Note that |
| | 84 | /// overcapacity can be negative. A simple way to have negative overcapacity is to |
| | 85 | /// allocate a fresh atom, which will have ideal capacity, and then grow it |
| | 86 | /// by 1 byte. It will then have -1 overcapacity. |
| | 87 | free_list: std.ArrayListUnmanaged(*Atom) = .{}, |
| | 88 | }; |
| | 89 | |
| 63 | base: File, | 90 | base: File, |
| 64 | | 91 | |
| 65 | /// If this is not null, an object file is created by LLVM and linked with LLD afterwards. | 92 | /// If this is not null, an object file is created by LLVM and linked with LLD afterwards. |
| ... | @@ -77,81 +104,32 @@ page_size: u16, | ... | @@ -77,81 +104,32 @@ page_size: u16, |
| 77 | /// fashion (default for LLVM backend). | 104 | /// fashion (default for LLVM backend). |
| 78 | mode: enum { incremental, one_shot }, | 105 | mode: enum { incremental, one_shot }, |
| 79 | | 106 | |
| 80 | /// The absolute address of the entry point. | 107 | uuid: macho.uuid_command = .{ |
| 81 | entry_addr: ?u64 = null, | 108 | .cmdsize = @sizeOf(macho.uuid_command), |
| 82 | | 109 | .uuid = undefined, |
| 83 | /// Code signature (if any) | 110 | }, |
| 84 | code_signature: ?CodeSignature = null, | | |
| 85 | | 111 | |
| 86 | objects: std.ArrayListUnmanaged(Object) = .{}, | 112 | objects: std.ArrayListUnmanaged(Object) = .{}, |
| 87 | archives: std.ArrayListUnmanaged(Archive) = .{}, | 113 | archives: std.ArrayListUnmanaged(Archive) = .{}, |
| 88 | | | |
| 89 | dylibs: std.ArrayListUnmanaged(Dylib) = .{}, | 114 | dylibs: std.ArrayListUnmanaged(Dylib) = .{}, |
| 90 | dylibs_map: std.StringHashMapUnmanaged(u16) = .{}, | 115 | dylibs_map: std.StringHashMapUnmanaged(u16) = .{}, |
| 91 | referenced_dylibs: std.AutoArrayHashMapUnmanaged(u16, void) = .{}, | 116 | referenced_dylibs: std.AutoArrayHashMapUnmanaged(u16, void) = .{}, |
| 92 | | 117 | |
| 93 | load_commands: std.ArrayListUnmanaged(macho.LoadCommand) = .{}, | 118 | segments: std.ArrayListUnmanaged(macho.segment_command_64) = .{}, |
| 94 | | 119 | sections: std.MultiArrayList(Section) = .{}, |
| 95 | pagezero_segment_cmd_index: ?u16 = null, | 120 | |
| 96 | text_segment_cmd_index: ?u16 = null, | 121 | pagezero_segment_cmd_index: ?u8 = null, |
| 97 | data_const_segment_cmd_index: ?u16 = null, | 122 | text_segment_cmd_index: ?u8 = null, |
| 98 | data_segment_cmd_index: ?u16 = null, | 123 | data_const_segment_cmd_index: ?u8 = null, |
| 99 | linkedit_segment_cmd_index: ?u16 = null, | 124 | data_segment_cmd_index: ?u8 = null, |
| 100 | dyld_info_cmd_index: ?u16 = null, | 125 | linkedit_segment_cmd_index: ?u8 = null, |
| 101 | symtab_cmd_index: ?u16 = null, | 126 | |
| 102 | dysymtab_cmd_index: ?u16 = null, | 127 | text_section_index: ?u8 = null, |
| 103 | dylinker_cmd_index: ?u16 = null, | 128 | stubs_section_index: ?u8 = null, |
| 104 | data_in_code_cmd_index: ?u16 = null, | 129 | stub_helper_section_index: ?u8 = null, |
| 105 | function_starts_cmd_index: ?u16 = null, | 130 | got_section_index: ?u8 = null, |
| 106 | main_cmd_index: ?u16 = null, | 131 | la_symbol_ptr_section_index: ?u8 = null, |
| 107 | dylib_id_cmd_index: ?u16 = null, | 132 | data_section_index: ?u8 = null, |
| 108 | source_version_cmd_index: ?u16 = null, | | |
| 109 | build_version_cmd_index: ?u16 = null, | | |
| 110 | uuid_cmd_index: ?u16 = null, | | |
| 111 | code_signature_cmd_index: ?u16 = null, | | |
| 112 | | | |
| 113 | // __TEXT segment sections | | |
| 114 | text_section_index: ?u16 = null, | | |
| 115 | stubs_section_index: ?u16 = null, | | |
| 116 | stub_helper_section_index: ?u16 = null, | | |
| 117 | text_const_section_index: ?u16 = null, | | |
| 118 | cstring_section_index: ?u16 = null, | | |
| 119 | ustring_section_index: ?u16 = null, | | |
| 120 | gcc_except_tab_section_index: ?u16 = null, | | |
| 121 | unwind_info_section_index: ?u16 = null, | | |
| 122 | eh_frame_section_index: ?u16 = null, | | |
| 123 | | | |
| 124 | objc_methlist_section_index: ?u16 = null, | | |
| 125 | objc_methname_section_index: ?u16 = null, | | |
| 126 | objc_methtype_section_index: ?u16 = null, | | |
| 127 | objc_classname_section_index: ?u16 = null, | | |
| 128 | | | |
| 129 | // __DATA_CONST segment sections | | |
| 130 | got_section_index: ?u16 = null, | | |
| 131 | mod_init_func_section_index: ?u16 = null, | | |
| 132 | mod_term_func_section_index: ?u16 = null, | | |
| 133 | data_const_section_index: ?u16 = null, | | |
| 134 | | | |
| 135 | objc_cfstring_section_index: ?u16 = null, | | |
| 136 | objc_classlist_section_index: ?u16 = null, | | |
| 137 | objc_imageinfo_section_index: ?u16 = null, | | |
| 138 | | | |
| 139 | // __DATA segment sections | | |
| 140 | tlv_section_index: ?u16 = null, | | |
| 141 | tlv_data_section_index: ?u16 = null, | | |
| 142 | tlv_bss_section_index: ?u16 = null, | | |
| 143 | tlv_ptrs_section_index: ?u16 = null, | | |
| 144 | la_symbol_ptr_section_index: ?u16 = null, | | |
| 145 | data_section_index: ?u16 = null, | | |
| 146 | bss_section_index: ?u16 = null, | | |
| 147 | | | |
| 148 | objc_const_section_index: ?u16 = null, | | |
| 149 | objc_selrefs_section_index: ?u16 = null, | | |
| 150 | objc_classrefs_section_index: ?u16 = null, | | |
| 151 | objc_data_section_index: ?u16 = null, | | |
| 152 | | | |
| 153 | rustc_section_index: ?u16 = null, | | |
| 154 | rustc_section_size: u64 = 0, | | |
| 155 | | 133 | |
| 156 | locals: std.ArrayListUnmanaged(macho.nlist_64) = .{}, | 134 | locals: std.ArrayListUnmanaged(macho.nlist_64) = .{}, |
| 157 | globals: std.StringArrayHashMapUnmanaged(SymbolWithLoc) = .{}, | 135 | globals: std.StringArrayHashMapUnmanaged(SymbolWithLoc) = .{}, |
| ... | @@ -188,37 +166,12 @@ stubs_table: std.AutoHashMapUnmanaged(SymbolWithLoc, u32) = .{}, | ... | @@ -188,37 +166,12 @@ stubs_table: std.AutoHashMapUnmanaged(SymbolWithLoc, u32) = .{}, |
| 188 | | 166 | |
| 189 | error_flags: File.ErrorFlags = File.ErrorFlags{}, | 167 | error_flags: File.ErrorFlags = File.ErrorFlags{}, |
| 190 | | 168 | |
| 191 | load_commands_dirty: bool = false, | | |
| 192 | sections_order_dirty: bool = false, | | |
| 193 | | | |
| 194 | /// A helper var to indicate if we are at the start of the incremental updates, or | 169 | /// A helper var to indicate if we are at the start of the incremental updates, or |
| 195 | /// already somewhere further along the update-and-run chain. | 170 | /// already somewhere further along the update-and-run chain. |
| 196 | /// TODO once we add opening a prelinked output binary from file, this will become | 171 | /// TODO once we add opening a prelinked output binary from file, this will become |
| 197 | /// obsolete as we will carry on where we left off. | 172 | /// obsolete as we will carry on where we left off. |
| 198 | cold_start: bool = true, | 173 | cold_start: bool = true, |
| 199 | | 174 | |
| 200 | section_ordinals: std.AutoArrayHashMapUnmanaged(MatchingSection, void) = .{}, | | |
| 201 | | | |
| 202 | /// A list of atoms that have surplus capacity. This list can have false | | |
| 203 | /// positives, as functions grow and shrink over time, only sometimes being added | | |
| 204 | /// or removed from the freelist. | | |
| 205 | /// | | |
| 206 | /// An atom has surplus capacity when its overcapacity value is greater than | | |
| 207 | /// padToIdeal(minimum_atom_size). That is, when it has so | | |
| 208 | /// much extra capacity, that we could fit a small new symbol in it, itself with | | |
| 209 | /// ideal_capacity or more. | | |
| 210 | /// | | |
| 211 | /// Ideal capacity is defined by size + (size / ideal_factor). | | |
| 212 | /// | | |
| 213 | /// Overcapacity is measured by actual_capacity - ideal_capacity. Note that | | |
| 214 | /// overcapacity can be negative. A simple way to have negative overcapacity is to | | |
| 215 | /// allocate a fresh atom, which will have ideal capacity, and then grow it | | |
| 216 | /// by 1 byte. It will then have -1 overcapacity. | | |
| 217 | atom_free_lists: std.AutoHashMapUnmanaged(MatchingSection, std.ArrayListUnmanaged(*Atom)) = .{}, | | |
| 218 | | | |
| 219 | /// Pointer to the last allocated atom | | |
| 220 | atoms: std.AutoHashMapUnmanaged(MatchingSection, *Atom) = .{}, | | |
| 221 | | | |
| 222 | /// List of atoms that are either synthetic or map directly to the Zig source program. | 175 | /// List of atoms that are either synthetic or map directly to the Zig source program. |
| 223 | managed_atoms: std.ArrayListUnmanaged(*Atom) = .{}, | 176 | managed_atoms: std.ArrayListUnmanaged(*Atom) = .{}, |
| 224 | | 177 | |
| ... | @@ -250,7 +203,7 @@ unnamed_const_atoms: UnnamedConstTable = .{}, | ... | @@ -250,7 +203,7 @@ unnamed_const_atoms: UnnamedConstTable = .{}, |
| 250 | /// We store them here so that we can properly dispose of any allocated | 203 | /// We store them here so that we can properly dispose of any allocated |
| 251 | /// memory within the atom in the incremental linker. | 204 | /// memory within the atom in the incremental linker. |
| 252 | /// TODO consolidate this. | 205 | /// TODO consolidate this. |
| 253 | decls: std.AutoArrayHashMapUnmanaged(Module.Decl.Index, ?MatchingSection) = .{}, | 206 | decls: std.AutoArrayHashMapUnmanaged(Module.Decl.Index, ?u8) = .{}, |
| 254 | | 207 | |
| 255 | const Entry = struct { | 208 | const Entry = struct { |
| 256 | target: SymbolWithLoc, | 209 | target: SymbolWithLoc, |
| ... | @@ -408,12 +361,7 @@ pub fn openPath(allocator: Allocator, options: link.Options) !*MachO { | ... | @@ -408,12 +361,7 @@ pub fn openPath(allocator: Allocator, options: link.Options) !*MachO { |
| 408 | | 361 | |
| 409 | pub fn createEmpty(gpa: Allocator, options: link.Options) !*MachO { | 362 | pub fn createEmpty(gpa: Allocator, options: link.Options) !*MachO { |
| 410 | const cpu_arch = options.target.cpu.arch; | 363 | const cpu_arch = options.target.cpu.arch; |
| 411 | const os_tag = options.target.os.tag; | | |
| 412 | const abi = options.target.abi; | | |
| 413 | const page_size: u16 = if (cpu_arch == .aarch64) 0x4000 else 0x1000; | 364 | const page_size: u16 = if (cpu_arch == .aarch64) 0x4000 else 0x1000; |
| 414 | // Adhoc code signature is required when targeting aarch64-macos either directly or indirectly via the simulator | | |
| 415 | // ABI such as aarch64-ios-simulator, etc. | | |
| 416 | const requires_adhoc_codesig = cpu_arch == .aarch64 and (os_tag == .macos or abi == .simulator); | | |
| 417 | const use_llvm = build_options.have_llvm and options.use_llvm; | 365 | const use_llvm = build_options.have_llvm and options.use_llvm; |
| 418 | const use_stage1 = build_options.is_stage1 and options.use_stage1; | 366 | const use_stage1 = build_options.is_stage1 and options.use_stage1; |
| 419 | | 367 | |
| ... | @@ -428,10 +376,6 @@ pub fn createEmpty(gpa: Allocator, options: link.Options) !*MachO { | ... | @@ -428,10 +376,6 @@ pub fn createEmpty(gpa: Allocator, options: link.Options) !*MachO { |
| 428 | .file = null, | 376 | .file = null, |
| 429 | }, | 377 | }, |
| 430 | .page_size = page_size, | 378 | .page_size = page_size, |
| 431 | .code_signature = if (requires_adhoc_codesig) | | |
| 432 | CodeSignature.init(page_size) | | |
| 433 | else | | |
| 434 | null, | | |
| 435 | .mode = if (use_stage1 or use_llvm or options.module == null or options.cache_mode == .whole) | 379 | .mode = if (use_stage1 or use_llvm or options.module == null or options.cache_mode == .whole) |
| 436 | .one_shot | 380 | .one_shot |
| 437 | else | 381 | else |
| ... | @@ -562,8 +506,8 @@ pub fn flushModule(self: *MachO, comp: *Compilation, prog_node: *std.Progress.No | ... | @@ -562,8 +506,8 @@ pub fn flushModule(self: *MachO, comp: *Compilation, prog_node: *std.Progress.No |
| 562 | var dependent_libs = std.fifo.LinearFifo(struct { | 506 | var dependent_libs = std.fifo.LinearFifo(struct { |
| 563 | id: Dylib.Id, | 507 | id: Dylib.Id, |
| 564 | parent: u16, | 508 | parent: u16, |
| 565 | }, .Dynamic).init(self.base.allocator); | 509 | }, .Dynamic).init(arena); |
| 566 | defer dependent_libs.deinit(); | 510 | |
| 567 | try self.parseLibs(libs.keys(), libs.values(), self.base.options.sysroot, &dependent_libs); | 511 | try self.parseLibs(libs.keys(), libs.values(), self.base.options.sysroot, &dependent_libs); |
| 568 | try self.parseDependentLibs(self.base.options.sysroot, &dependent_libs); | 512 | try self.parseDependentLibs(self.base.options.sysroot, &dependent_libs); |
| 569 | } | 513 | } |
| ... | @@ -573,7 +517,6 @@ pub fn flushModule(self: *MachO, comp: *Compilation, prog_node: *std.Progress.No | ... | @@ -573,7 +517,6 @@ pub fn flushModule(self: *MachO, comp: *Compilation, prog_node: *std.Progress.No |
| 573 | try self.createDyldPrivateAtom(); | 517 | try self.createDyldPrivateAtom(); |
| 574 | try self.createStubHelperPreambleAtom(); | 518 | try self.createStubHelperPreambleAtom(); |
| 575 | try self.resolveSymbolsInDylibs(); | 519 | try self.resolveSymbolsInDylibs(); |
| 576 | try self.addCodeSignatureLC(); | | |
| 577 | | 520 | |
| 578 | if (self.unresolved.count() > 0) { | 521 | if (self.unresolved.count() > 0) { |
| 579 | return error.UndefinedSymbolReference; | 522 | return error.UndefinedSymbolReference; |
| ... | @@ -583,67 +526,92 @@ pub fn flushModule(self: *MachO, comp: *Compilation, prog_node: *std.Progress.No | ... | @@ -583,67 +526,92 @@ pub fn flushModule(self: *MachO, comp: *Compilation, prog_node: *std.Progress.No |
| 583 | | 526 | |
| 584 | if (build_options.enable_logging) { | 527 | if (build_options.enable_logging) { |
| 585 | self.logSymtab(); | 528 | self.logSymtab(); |
| 586 | self.logSectionOrdinals(); | 529 | self.logSections(); |
| 587 | self.logAtoms(); | 530 | self.logAtoms(); |
| 588 | } | 531 | } |
| 589 | | 532 | |
| 590 | try self.writeAtomsIncremental(); | 533 | try self.writeAtomsIncremental(); |
| 591 | | 534 | |
| 592 | try self.setEntryPoint(); | 535 | var lc_buffer = std.ArrayList(u8).init(arena); |
| 593 | try self.updateSectionOrdinals(); | 536 | const lc_writer = lc_buffer.writer(); |
| 594 | try self.writeLinkeditSegment(); | 537 | var ncmds: u32 = 0; |
| 595 | | 538 | |
| 596 | if (self.d_sym) |*d_sym| { | 539 | try self.writeLinkeditSegmentData(&ncmds, lc_writer); |
| 597 | // Flush debug symbols bundle. | 540 | try writeDylinkerLC(&ncmds, lc_writer); |
| 598 | try d_sym.flushModule(self.base.allocator, self.base.options); | 541 | |
| | 542 | self.writeMainLC(&ncmds, lc_writer) catch |err| switch (err) { |
| | 543 | error.MissingMainEntrypoint => { |
| | 544 | self.error_flags.no_entry_point_found = true; |
| | 545 | }, |
| | 546 | else => |e| return e, |
| | 547 | }; |
| | 548 | |
| | 549 | try self.writeDylibIdLC(&ncmds, lc_writer); |
| | 550 | try self.writeRpathLCs(&ncmds, lc_writer); |
| | 551 | |
| | 552 | { |
| | 553 | try lc_writer.writeStruct(macho.source_version_command{ |
| | 554 | .cmdsize = @sizeOf(macho.source_version_command), |
| | 555 | .version = 0x0, |
| | 556 | }); |
| | 557 | ncmds += 1; |
| 599 | } | 558 | } |
| 600 | | 559 | |
| 601 | // code signature and entitlements | 560 | try self.writeBuildVersionLC(&ncmds, lc_writer); |
| 602 | if (self.base.options.entitlements) |path| { | 561 | |
| 603 | if (self.code_signature) |*csig| { | 562 | { |
| 604 | try csig.addEntitlements(self.base.allocator, path); | 563 | std.crypto.random.bytes(&self.uuid.uuid); |
| 605 | csig.code_directory.ident = self.base.options.emit.?.sub_path; | 564 | try lc_writer.writeStruct(self.uuid); |
| 606 | } else { | 565 | ncmds += 1; |
| 607 | var csig = CodeSignature.init(self.page_size); | | |
| 608 | try csig.addEntitlements(self.base.allocator, path); | | |
| 609 | csig.code_directory.ident = self.base.options.emit.?.sub_path; | | |
| 610 | self.code_signature = csig; | | |
| 611 | } | | |
| 612 | } | 566 | } |
| 613 | | 567 | |
| 614 | if (self.code_signature) |*csig| { | 568 | try self.writeLoadDylibLCs(&ncmds, lc_writer); |
| 615 | csig.clear(self.base.allocator); | 569 | |
| 616 | csig.code_directory.ident = self.base.options.emit.?.sub_path; | 570 | const target = self.base.options.target; |
| | 571 | const requires_codesig = blk: { |
| | 572 | if (self.base.options.entitlements) |_| break :blk true; |
| | 573 | if (target.cpu.arch == .aarch64 and (target.os.tag == .macos or target.abi == .simulator)) |
| | 574 | break :blk true; |
| | 575 | break :blk false; |
| | 576 | }; |
| | 577 | var codesig_offset: ?u32 = null; |
| | 578 | var codesig: ?CodeSignature = if (requires_codesig) blk: { |
| 617 | // Preallocate space for the code signature. | 579 | // Preallocate space for the code signature. |
| 618 | // We need to do this at this stage so that we have the load commands with proper values | 580 | // We need to do this at this stage so that we have the load commands with proper values |
| 619 | // written out to the file. | 581 | // written out to the file. |
| 620 | // The most important here is to have the correct vm and filesize of the __LINKEDIT segment | 582 | // The most important here is to have the correct vm and filesize of the __LINKEDIT segment |
| 621 | // where the code signature goes into. | 583 | // where the code signature goes into. |
| 622 | try self.writeCodeSignaturePadding(csig); | 584 | var codesig = CodeSignature.init(self.page_size); |
| 623 | } | 585 | codesig.code_directory.ident = self.base.options.emit.?.sub_path; |
| | 586 | if (self.base.options.entitlements) |path| { |
| | 587 | try codesig.addEntitlements(arena, path); |
| | 588 | } |
| | 589 | codesig_offset = try self.writeCodeSignaturePadding(&codesig, &ncmds, lc_writer); |
| | 590 | break :blk codesig; |
| | 591 | } else null; |
| 624 | | 592 | |
| 625 | try self.writeLoadCommands(); | 593 | var headers_buf = std.ArrayList(u8).init(arena); |
| 626 | try self.writeHeader(); | 594 | try self.writeSegmentHeaders(&ncmds, headers_buf.writer()); |
| 627 | | 595 | |
| 628 | if (self.entry_addr == null and self.base.options.output_mode == .Exe) { | 596 | try self.base.file.?.pwriteAll(headers_buf.items, @sizeOf(macho.mach_header_64)); |
| 629 | log.debug("flushing. no_entry_point_found = true", .{}); | 597 | try self.base.file.?.pwriteAll(lc_buffer.items, @sizeOf(macho.mach_header_64) + headers_buf.items.len); |
| 630 | self.error_flags.no_entry_point_found = true; | | |
| 631 | } else { | | |
| 632 | log.debug("flushing. no_entry_point_found = false", .{}); | | |
| 633 | self.error_flags.no_entry_point_found = false; | | |
| 634 | } | | |
| 635 | | 598 | |
| 636 | assert(!self.load_commands_dirty); | 599 | try self.writeHeader(ncmds, @intCast(u32, lc_buffer.items.len + headers_buf.items.len)); |
| 637 | | 600 | |
| 638 | if (self.code_signature) |*csig| { | 601 | if (codesig) |*csig| { |
| 639 | try self.writeCodeSignature(csig); // code signing always comes last | 602 | try self.writeCodeSignature(csig, codesig_offset.?); // code signing always comes last |
| 640 | } | 603 | } |
| 641 | | 604 | |
| 642 | if (build_options.enable_link_snapshots) { | 605 | if (self.d_sym) |*d_sym| { |
| 643 | if (self.base.options.enable_link_snapshots) | 606 | // Flush debug symbols bundle. |
| 644 | try self.snapshotState(); | 607 | try d_sym.flushModule(self.base.allocator, self.base.options); |
| 645 | } | 608 | } |
| 646 | | 609 | |
| | 610 | // if (build_options.enable_link_snapshots) { |
| | 611 | // if (self.base.options.enable_link_snapshots) |
| | 612 | // try self.snapshotState(); |
| | 613 | // } |
| | 614 | |
| 647 | if (cache_miss) { | 615 | if (cache_miss) { |
| 648 | // Update the file with the digest. If it fails we can continue; it only | 616 | // Update the file with the digest. If it fails we can continue; it only |
| 649 | // means that the next invocation will have an unnecessary cache miss. | 617 | // means that the next invocation will have an unnecessary cache miss. |
| ... | @@ -708,6 +676,9 @@ fn linkOneShot(self: *MachO, comp: *Compilation, prog_node: *std.Progress.Node) | ... | @@ -708,6 +676,9 @@ fn linkOneShot(self: *MachO, comp: *Compilation, prog_node: *std.Progress.Node) |
| 708 | sub_prog_node.context.refresh(); | 676 | sub_prog_node.context.refresh(); |
| 709 | defer sub_prog_node.end(); | 677 | defer sub_prog_node.end(); |
| 710 | | 678 | |
| | 679 | const cpu_arch = self.base.options.target.cpu.arch; |
| | 680 | const os_tag = self.base.options.target.os.tag; |
| | 681 | const abi = self.base.options.target.abi; |
| 711 | const is_lib = self.base.options.output_mode == .Lib; | 682 | const is_lib = self.base.options.output_mode == .Lib; |
| 712 | const is_dyn_lib = self.base.options.link_mode == .Dynamic and is_lib; | 683 | const is_dyn_lib = self.base.options.link_mode == .Dynamic and is_lib; |
| 713 | const is_exe_or_dyn_lib = is_dyn_lib or self.base.options.output_mode == .Exe; | 684 | const is_exe_or_dyn_lib = is_dyn_lib or self.base.options.output_mode == .Exe; |
| ... | @@ -990,40 +961,6 @@ fn linkOneShot(self: *MachO, comp: *Compilation, prog_node: *std.Progress.Node) | ... | @@ -990,40 +961,6 @@ fn linkOneShot(self: *MachO, comp: *Compilation, prog_node: *std.Progress.Node) |
| 990 | } | 961 | } |
| 991 | } | 962 | } |
| 992 | | 963 | |
| 993 | // rpaths | | |
| 994 | var rpath_table = std.StringArrayHashMap(void).init(arena); | | |
| 995 | for (self.base.options.rpath_list) |rpath| { | | |
| 996 | if (rpath_table.contains(rpath)) continue; | | |
| 997 | const cmdsize = @intCast(u32, mem.alignForwardGeneric( | | |
| 998 | u64, | | |
| 999 | @sizeOf(macho.rpath_command) + rpath.len + 1, | | |
| 1000 | @sizeOf(u64), | | |
| 1001 | )); | | |
| 1002 | var rpath_cmd = macho.emptyGenericCommandWithData(macho.rpath_command{ | | |
| 1003 | .cmdsize = cmdsize, | | |
| 1004 | .path = @sizeOf(macho.rpath_command), | | |
| 1005 | }); | | |
| 1006 | rpath_cmd.data = try gpa.alloc(u8, cmdsize - rpath_cmd.inner.path); | | |
| 1007 | mem.set(u8, rpath_cmd.data, 0); | | |
| 1008 | mem.copy(u8, rpath_cmd.data, rpath); | | |
| 1009 | try self.load_commands.append(gpa, .{ .rpath = rpath_cmd }); | | |
| 1010 | try rpath_table.putNoClobber(rpath, {}); | | |
| 1011 | self.load_commands_dirty = true; | | |
| 1012 | } | | |
| 1013 | | | |
| 1014 | // code signature and entitlements | | |
| 1015 | if (self.base.options.entitlements) |path| { | | |
| 1016 | if (self.code_signature) |*csig| { | | |
| 1017 | try csig.addEntitlements(gpa, path); | | |
| 1018 | csig.code_directory.ident = self.base.options.emit.?.sub_path; | | |
| 1019 | } else { | | |
| 1020 | var csig = CodeSignature.init(self.page_size); | | |
| 1021 | try csig.addEntitlements(gpa, path); | | |
| 1022 | csig.code_directory.ident = self.base.options.emit.?.sub_path; | | |
| 1023 | self.code_signature = csig; | | |
| 1024 | } | | |
| 1025 | } | | |
| 1026 | | | |
| 1027 | if (self.base.options.verbose_link) { | 964 | if (self.base.options.verbose_link) { |
| 1028 | var argv = std.ArrayList([]const u8).init(arena); | 965 | var argv = std.ArrayList([]const u8).init(arena); |
| 1029 | | 966 | |
| ... | @@ -1048,7 +985,7 @@ fn linkOneShot(self: *MachO, comp: *Compilation, prog_node: *std.Progress.Node) | ... | @@ -1048,7 +985,7 @@ fn linkOneShot(self: *MachO, comp: *Compilation, prog_node: *std.Progress.Node) |
| 1048 | try argv.append(syslibroot); | 985 | try argv.append(syslibroot); |
| 1049 | } | 986 | } |
| 1050 | | 987 | |
| 1051 | for (rpath_table.keys()) |rpath| { | 988 | for (self.base.options.rpath_list) |rpath| { |
| 1052 | try argv.append("-rpath"); | 989 | try argv.append("-rpath"); |
| 1053 | try argv.append(rpath); | 990 | try argv.append(rpath); |
| 1054 | } | 991 | } |
| ... | @@ -1157,25 +1094,22 @@ fn linkOneShot(self: *MachO, comp: *Compilation, prog_node: *std.Progress.Node) | ... | @@ -1157,25 +1094,22 @@ fn linkOneShot(self: *MachO, comp: *Compilation, prog_node: *std.Progress.Node) |
| 1157 | var dependent_libs = std.fifo.LinearFifo(struct { | 1094 | var dependent_libs = std.fifo.LinearFifo(struct { |
| 1158 | id: Dylib.Id, | 1095 | id: Dylib.Id, |
| 1159 | parent: u16, | 1096 | parent: u16, |
| 1160 | }, .Dynamic).init(gpa); | 1097 | }, .Dynamic).init(arena); |
| 1161 | defer dependent_libs.deinit(); | 1098 | |
| 1162 | try self.parseInputFiles(positionals.items, self.base.options.sysroot, &dependent_libs); | 1099 | try self.parseInputFiles(positionals.items, self.base.options.sysroot, &dependent_libs); |
| 1163 | try self.parseAndForceLoadStaticArchives(must_link_archives.keys()); | 1100 | try self.parseAndForceLoadStaticArchives(must_link_archives.keys()); |
| 1164 | try self.parseLibs(libs.keys(), libs.values(), self.base.options.sysroot, &dependent_libs); | 1101 | try self.parseLibs(libs.keys(), libs.values(), self.base.options.sysroot, &dependent_libs); |
| 1165 | try self.parseDependentLibs(self.base.options.sysroot, &dependent_libs); | 1102 | try self.parseDependentLibs(self.base.options.sysroot, &dependent_libs); |
| 1166 | | 1103 | |
| 1167 | for (self.objects.items) |*object, object_id| { | 1104 | for (self.objects.items) |_, object_id| { |
| 1168 | try self.resolveSymbolsInObject(object, @intCast(u16, object_id)); | 1105 | try self.resolveSymbolsInObject(@intCast(u16, object_id)); |
| 1169 | } | 1106 | } |
| 1170 | | 1107 | |
| 1171 | try self.resolveSymbolsInArchives(); | 1108 | try self.resolveSymbolsInArchives(); |
| 1172 | try self.resolveDyldStubBinder(); | 1109 | try self.resolveDyldStubBinder(); |
| 1173 | try self.createDyldPrivateAtom(); | | |
| 1174 | try self.createStubHelperPreambleAtom(); | | |
| 1175 | try self.resolveSymbolsInDylibs(); | 1110 | try self.resolveSymbolsInDylibs(); |
| 1176 | try self.createMhExecuteHeaderSymbol(); | 1111 | try self.createMhExecuteHeaderSymbol(); |
| 1177 | try self.createDsoHandleSymbol(); | 1112 | try self.createDsoHandleSymbol(); |
| 1178 | try self.addCodeSignatureLC(); | | |
| 1179 | try self.resolveSymbolsAtLoading(); | 1113 | try self.resolveSymbolsAtLoading(); |
| 1180 | | 1114 | |
| 1181 | if (self.unresolved.count() > 0) { | 1115 | if (self.unresolved.count() > 0) { |
| ... | @@ -1188,7 +1122,13 @@ fn linkOneShot(self: *MachO, comp: *Compilation, prog_node: *std.Progress.Node) | ... | @@ -1188,7 +1122,13 @@ fn linkOneShot(self: *MachO, comp: *Compilation, prog_node: *std.Progress.Node) |
| 1188 | return error.FrameworkNotFound; | 1122 | return error.FrameworkNotFound; |
| 1189 | } | 1123 | } |
| 1190 | | 1124 | |
| | 1125 | for (self.objects.items) |*object| { |
| | 1126 | try object.scanInputSections(self); |
| | 1127 | } |
| | 1128 | |
| | 1129 | try self.createDyldPrivateAtom(); |
| 1191 | try self.createTentativeDefAtoms(); | 1130 | try self.createTentativeDefAtoms(); |
| | 1131 | try self.createStubHelperPreambleAtom(); |
| 1192 | | 1132 | |
| 1193 | for (self.objects.items) |*object, object_id| { | 1133 | for (self.objects.items) |*object, object_id| { |
| 1194 | try object.splitIntoAtomsOneShot(self, @intCast(u32, object_id)); | 1134 | try object.splitIntoAtomsOneShot(self, @intCast(u32, object_id)); |
| ... | @@ -1198,49 +1138,82 @@ fn linkOneShot(self: *MachO, comp: *Compilation, prog_node: *std.Progress.Node) | ... | @@ -1198,49 +1138,82 @@ fn linkOneShot(self: *MachO, comp: *Compilation, prog_node: *std.Progress.Node) |
| 1198 | try dead_strip.gcAtoms(self); | 1138 | try dead_strip.gcAtoms(self); |
| 1199 | } | 1139 | } |
| 1200 | | 1140 | |
| 1201 | try self.pruneAndSortSections(); | | |
| 1202 | try self.allocateSegments(); | 1141 | try self.allocateSegments(); |
| 1203 | try self.allocateSymbols(); | 1142 | try self.allocateSymbols(); |
| 1204 | | 1143 | |
| 1205 | try self.allocateSpecialSymbols(); | 1144 | try self.allocateSpecialSymbols(); |
| 1206 | | 1145 | |
| 1207 | if (build_options.enable_logging) { | 1146 | if (build_options.enable_logging or true) { |
| 1208 | self.logSymtab(); | 1147 | self.logSymtab(); |
| 1209 | self.logSectionOrdinals(); | 1148 | self.logSections(); |
| 1210 | self.logAtoms(); | 1149 | self.logAtoms(); |
| 1211 | } | 1150 | } |
| 1212 | | 1151 | |
| 1213 | try self.writeAtomsOneShot(); | 1152 | try self.writeAtomsOneShot(); |
| 1214 | | 1153 | |
| 1215 | if (self.rustc_section_index) |id| { | 1154 | var lc_buffer = std.ArrayList(u8).init(arena); |
| 1216 | const sect = self.getSectionPtr(.{ | 1155 | const lc_writer = lc_buffer.writer(); |
| 1217 | .seg = self.data_segment_cmd_index.?, | 1156 | var ncmds: u32 = 0; |
| 1218 | .sect = id, | 1157 | |
| | 1158 | try self.writeLinkeditSegmentData(&ncmds, lc_writer); |
| | 1159 | try writeDylinkerLC(&ncmds, lc_writer); |
| | 1160 | try self.writeMainLC(&ncmds, lc_writer); |
| | 1161 | try self.writeDylibIdLC(&ncmds, lc_writer); |
| | 1162 | try self.writeRpathLCs(&ncmds, lc_writer); |
| | 1163 | |
| | 1164 | { |
| | 1165 | try lc_writer.writeStruct(macho.source_version_command{ |
| | 1166 | .cmdsize = @sizeOf(macho.source_version_command), |
| | 1167 | .version = 0x0, |
| 1219 | }); | 1168 | }); |
| 1220 | sect.size = self.rustc_section_size; | 1169 | ncmds += 1; |
| 1221 | } | 1170 | } |
| 1222 | | 1171 | |
| 1223 | try self.setEntryPoint(); | 1172 | try self.writeBuildVersionLC(&ncmds, lc_writer); |
| 1224 | try self.writeLinkeditSegment(); | | |
| 1225 | | 1173 | |
| 1226 | if (self.code_signature) |*csig| { | 1174 | { |
| 1227 | csig.clear(gpa); | 1175 | var uuid_lc = macho.uuid_command{ |
| 1228 | csig.code_directory.ident = self.base.options.emit.?.sub_path; | 1176 | .cmdsize = @sizeOf(macho.uuid_command), |
| | 1177 | .uuid = undefined, |
| | 1178 | }; |
| | 1179 | std.crypto.random.bytes(&uuid_lc.uuid); |
| | 1180 | try lc_writer.writeStruct(uuid_lc); |
| | 1181 | ncmds += 1; |
| | 1182 | } |
| | 1183 | |
| | 1184 | try self.writeLoadDylibLCs(&ncmds, lc_writer); |
| | 1185 | |
| | 1186 | const requires_codesig = blk: { |
| | 1187 | if (self.base.options.entitlements) |_| break :blk true; |
| | 1188 | if (cpu_arch == .aarch64 and (os_tag == .macos or abi == .simulator)) break :blk true; |
| | 1189 | break :blk false; |
| | 1190 | }; |
| | 1191 | var codesig_offset: ?u32 = null; |
| | 1192 | var codesig: ?CodeSignature = if (requires_codesig) blk: { |
| 1229 | // Preallocate space for the code signature. | 1193 | // Preallocate space for the code signature. |
| 1230 | // We need to do this at this stage so that we have the load commands with proper values | 1194 | // We need to do this at this stage so that we have the load commands with proper values |
| 1231 | // written out to the file. | 1195 | // written out to the file. |
| 1232 | // The most important here is to have the correct vm and filesize of the __LINKEDIT segment | 1196 | // The most important here is to have the correct vm and filesize of the __LINKEDIT segment |
| 1233 | // where the code signature goes into. | 1197 | // where the code signature goes into. |
| 1234 | try self.writeCodeSignaturePadding(csig); | 1198 | var codesig = CodeSignature.init(self.page_size); |
| 1235 | } | 1199 | codesig.code_directory.ident = self.base.options.emit.?.sub_path; |
| | 1200 | if (self.base.options.entitlements) |path| { |
| | 1201 | try codesig.addEntitlements(arena, path); |
| | 1202 | } |
| | 1203 | codesig_offset = try self.writeCodeSignaturePadding(&codesig, &ncmds, lc_writer); |
| | 1204 | break :blk codesig; |
| | 1205 | } else null; |
| | 1206 | |
| | 1207 | var headers_buf = std.ArrayList(u8).init(arena); |
| | 1208 | try self.writeSegmentHeaders(&ncmds, headers_buf.writer()); |
| 1236 | | 1209 | |
| 1237 | try self.writeLoadCommands(); | 1210 | try self.base.file.?.pwriteAll(headers_buf.items, @sizeOf(macho.mach_header_64)); |
| 1238 | try self.writeHeader(); | 1211 | try self.base.file.?.pwriteAll(lc_buffer.items, @sizeOf(macho.mach_header_64) + headers_buf.items.len); |
| 1239 | | 1212 | |
| 1240 | assert(!self.load_commands_dirty); | 1213 | try self.writeHeader(ncmds, @intCast(u32, lc_buffer.items.len + headers_buf.items.len)); |
| 1241 | | 1214 | |
| 1242 | if (self.code_signature) |*csig| { | 1215 | if (codesig) |*csig| { |
| 1243 | try self.writeCodeSignature(csig); // code signing always comes last | 1216 | try self.writeCodeSignature(csig, codesig_offset.?); // code signing always comes last |
| 1244 | } | 1217 | } |
| 1245 | } | 1218 | } |
| 1246 | | 1219 | |
| ... | @@ -1395,66 +1368,77 @@ fn resolveFramework( | ... | @@ -1395,66 +1368,77 @@ fn resolveFramework( |
| 1395 | } | 1368 | } |
| 1396 | | 1369 | |
| 1397 | fn parseObject(self: *MachO, path: []const u8) !bool { | 1370 | fn parseObject(self: *MachO, path: []const u8) !bool { |
| | 1371 | const gpa = self.base.allocator; |
| 1398 | const file = fs.cwd().openFile(path, .{}) catch |err| switch (err) { | 1372 | const file = fs.cwd().openFile(path, .{}) catch |err| switch (err) { |
| 1399 | error.FileNotFound => return false, | 1373 | error.FileNotFound => return false, |
| 1400 | else => |e| return e, | 1374 | else => |e| return e, |
| 1401 | }; | 1375 | }; |
| 1402 | errdefer file.close(); | 1376 | defer file.close(); |
| 1403 | | | |
| 1404 | const name = try self.base.allocator.dupe(u8, path); | | |
| 1405 | errdefer self.base.allocator.free(name); | | |
| 1406 | | 1377 | |
| | 1378 | const name = try gpa.dupe(u8, path); |
| | 1379 | errdefer gpa.free(name); |
| | 1380 | const cpu_arch = self.base.options.target.cpu.arch; |
| 1407 | const mtime: u64 = mtime: { | 1381 | const mtime: u64 = mtime: { |
| 1408 | const stat = file.stat() catch break :mtime 0; | 1382 | const stat = file.stat() catch break :mtime 0; |
| 1409 | break :mtime @intCast(u64, @divFloor(stat.mtime, 1_000_000_000)); | 1383 | break :mtime @intCast(u64, @divFloor(stat.mtime, 1_000_000_000)); |
| 1410 | }; | 1384 | }; |
| | 1385 | const file_stat = try file.stat(); |
| | 1386 | const file_size = math.cast(usize, file_stat.size) orelse return error.Overflow; |
| | 1387 | const contents = try file.readToEndAllocOptions(gpa, file_size, file_size, @alignOf(u64), null); |
| 1411 | | 1388 | |
| 1412 | var object = Object{ | 1389 | var object = Object{ |
| 1413 | .name = name, | 1390 | .name = name, |
| 1414 | .file = file, | | |
| 1415 | .mtime = mtime, | 1391 | .mtime = mtime, |
| | 1392 | .contents = contents, |
| 1416 | }; | 1393 | }; |
| 1417 | | 1394 | |
| 1418 | object.parse(self.base.allocator, self.base.options.target.cpu.arch) catch |err| switch (err) { | 1395 | object.parse(gpa, cpu_arch) catch |err| switch (err) { |
| 1419 | error.EndOfStream, error.NotObject => { | 1396 | error.EndOfStream, error.NotObject => { |
| 1420 | object.deinit(self.base.allocator); | 1397 | object.deinit(gpa); |
| 1421 | return false; | 1398 | return false; |
| 1422 | }, | 1399 | }, |
| 1423 | else => |e| return e, | 1400 | else => |e| return e, |
| 1424 | }; | 1401 | }; |
| 1425 | | 1402 | |
| 1426 | try self.objects.append(self.base.allocator, object); | 1403 | try self.objects.append(gpa, object); |
| 1427 | | 1404 | |
| 1428 | return true; | 1405 | return true; |
| 1429 | } | 1406 | } |
| 1430 | | 1407 | |
| 1431 | fn parseArchive(self: *MachO, path: []const u8, force_load: bool) !bool { | 1408 | fn parseArchive(self: *MachO, path: []const u8, force_load: bool) !bool { |
| | 1409 | const gpa = self.base.allocator; |
| 1432 | const file = fs.cwd().openFile(path, .{}) catch |err| switch (err) { | 1410 | const file = fs.cwd().openFile(path, .{}) catch |err| switch (err) { |
| 1433 | error.FileNotFound => return false, | 1411 | error.FileNotFound => return false, |
| 1434 | else => |e| return e, | 1412 | else => |e| return e, |
| 1435 | }; | 1413 | }; |
| 1436 | errdefer file.close(); | 1414 | errdefer file.close(); |
| 1437 | | 1415 | |
| 1438 | const name = try self.base.allocator.dupe(u8, path); | 1416 | const name = try gpa.dupe(u8, path); |
| 1439 | errdefer self.base.allocator.free(name); | 1417 | errdefer gpa.free(name); |
| | 1418 | const cpu_arch = self.base.options.target.cpu.arch; |
| | 1419 | const reader = file.reader(); |
| | 1420 | const fat_offset = try fat.getLibraryOffset(reader, cpu_arch); |
| | 1421 | try reader.context.seekTo(fat_offset); |
| 1440 | | 1422 | |
| 1441 | var archive = Archive{ | 1423 | var archive = Archive{ |
| 1442 | .name = name, | 1424 | .name = name, |
| | 1425 | .fat_offset = fat_offset, |
| 1443 | .file = file, | 1426 | .file = file, |
| 1444 | }; | 1427 | }; |
| 1445 | | 1428 | |
| 1446 | archive.parse(self.base.allocator, self.base.options.target.cpu.arch) catch |err| switch (err) { | 1429 | archive.parse(gpa, reader) catch |err| switch (err) { |
| 1447 | error.EndOfStream, error.NotArchive => { | 1430 | error.EndOfStream, error.NotArchive => { |
| 1448 | archive.deinit(self.base.allocator); | 1431 | archive.deinit(gpa); |
| 1449 | return false; | 1432 | return false; |
| 1450 | }, | 1433 | }, |
| 1451 | else => |e| return e, | 1434 | else => |e| return e, |
| 1452 | }; | 1435 | }; |
| 1453 | | 1436 | |
| 1454 | if (force_load) { | 1437 | if (force_load) { |
| 1455 | defer archive.deinit(self.base.allocator); | 1438 | defer archive.deinit(gpa); |
| | 1439 | defer file.close(); |
| 1456 | // Get all offsets from the ToC | 1440 | // Get all offsets from the ToC |
| 1457 | var offsets = std.AutoArrayHashMap(u32, void).init(self.base.allocator); | 1441 | var offsets = std.AutoArrayHashMap(u32, void).init(gpa); |
| 1458 | defer offsets.deinit(); | 1442 | defer offsets.deinit(); |
| 1459 | for (archive.toc.values()) |offs| { | 1443 | for (archive.toc.values()) |offs| { |
| 1460 | for (offs.items) |off| { | 1444 | for (offs.items) |off| { |
| ... | @@ -1462,15 +1446,11 @@ fn parseArchive(self: *MachO, path: []const u8, force_load: bool) !bool { | ... | @@ -1462,15 +1446,11 @@ fn parseArchive(self: *MachO, path: []const u8, force_load: bool) !bool { |
| 1462 | } | 1446 | } |
| 1463 | } | 1447 | } |
| 1464 | for (offsets.keys()) |off| { | 1448 | for (offsets.keys()) |off| { |
| 1465 | const object = try self.objects.addOne(self.base.allocator); | 1449 | const object = try archive.parseObject(gpa, cpu_arch, off); |
| 1466 | object.* = try archive.parseObject( | 1450 | try self.objects.append(gpa, object); |
| 1467 | self.base.allocator, | | |
| 1468 | self.base.options.target.cpu.arch, | | |
| 1469 | off, | | |
| 1470 | ); | | |
| 1471 | } | 1451 | } |
| 1472 | } else { | 1452 | } else { |
| 1473 | try self.archives.append(self.base.allocator, archive); | 1453 | try self.archives.append(gpa, archive); |
| 1474 | } | 1454 | } |
| 1475 | | 1455 | |
| 1476 | return true; | 1456 | return true; |
| ... | @@ -1481,6 +1461,7 @@ const ParseDylibError = error{ | ... | @@ -1481,6 +1461,7 @@ const ParseDylibError = error{ |
| 1481 | EmptyStubFile, | 1461 | EmptyStubFile, |
| 1482 | MismatchedCpuArchitecture, | 1462 | MismatchedCpuArchitecture, |
| 1483 | UnsupportedCpuArchitecture, | 1463 | UnsupportedCpuArchitecture, |
| | 1464 | EndOfStream, |
| 1484 | } || fs.File.OpenError || std.os.PReadError || Dylib.Id.ParseError; | 1465 | } || fs.File.OpenError || std.os.PReadError || Dylib.Id.ParseError; |
| 1485 | | 1466 | |
| 1486 | const DylibCreateOpts = struct { | 1467 | const DylibCreateOpts = struct { |
| ... | @@ -1497,43 +1478,53 @@ pub fn parseDylib( | ... | @@ -1497,43 +1478,53 @@ pub fn parseDylib( |
| 1497 | dependent_libs: anytype, | 1478 | dependent_libs: anytype, |
| 1498 | opts: DylibCreateOpts, | 1479 | opts: DylibCreateOpts, |
| 1499 | ) ParseDylibError!bool { | 1480 | ) ParseDylibError!bool { |
| | 1481 | const gpa = self.base.allocator; |
| 1500 | const file = fs.cwd().openFile(path, .{}) catch |err| switch (err) { | 1482 | const file = fs.cwd().openFile(path, .{}) catch |err| switch (err) { |
| 1501 | error.FileNotFound => return false, | 1483 | error.FileNotFound => return false, |
| 1502 | else => |e| return e, | 1484 | else => |e| return e, |
| 1503 | }; | 1485 | }; |
| 1504 | errdefer file.close(); | 1486 | defer file.close(); |
| | 1487 | |
| | 1488 | const cpu_arch = self.base.options.target.cpu.arch; |
| | 1489 | const file_stat = try file.stat(); |
| | 1490 | var file_size = math.cast(usize, file_stat.size) orelse return error.Overflow; |
| 1505 | | 1491 | |
| 1506 | const name = try self.base.allocator.dupe(u8, path); | 1492 | const reader = file.reader(); |
| 1507 | errdefer self.base.allocator.free(name); | 1493 | const fat_offset = math.cast(usize, try fat.getLibraryOffset(reader, cpu_arch)) orelse |
| | 1494 | return error.Overflow; |
| | 1495 | try file.seekTo(fat_offset); |
| | 1496 | file_size -= fat_offset; |
| | 1497 | |
| | 1498 | const contents = try file.readToEndAllocOptions(gpa, file_size, file_size, @alignOf(u64), null); |
| | 1499 | defer gpa.free(contents); |
| 1508 | | 1500 | |
| 1509 | const dylib_id = @intCast(u16, self.dylibs.items.len); | 1501 | const dylib_id = @intCast(u16, self.dylibs.items.len); |
| 1510 | var dylib = Dylib{ | 1502 | var dylib = Dylib{ .weak = opts.weak }; |
| 1511 | .name = name, | | |
| 1512 | .file = file, | | |
| 1513 | .weak = opts.weak, | | |
| 1514 | }; | | |
| 1515 | | 1503 | |
| 1516 | dylib.parse( | 1504 | dylib.parseFromBinary( |
| 1517 | self.base.allocator, | 1505 | gpa, |
| 1518 | self.base.options.target.cpu.arch, | 1506 | cpu_arch, |
| 1519 | dylib_id, | 1507 | dylib_id, |
| 1520 | dependent_libs, | 1508 | dependent_libs, |
| | 1509 | path, |
| | 1510 | contents, |
| 1521 | ) catch |err| switch (err) { | 1511 | ) catch |err| switch (err) { |
| 1522 | error.EndOfStream, error.NotDylib => { | 1512 | error.EndOfStream, error.NotDylib => { |
| 1523 | try file.seekTo(0); | 1513 | try file.seekTo(0); |
| 1524 | | 1514 | |
| 1525 | var lib_stub = LibStub.loadFromFile(self.base.allocator, file) catch { | 1515 | var lib_stub = LibStub.loadFromFile(gpa, file) catch { |
| 1526 | dylib.deinit(self.base.allocator); | 1516 | dylib.deinit(gpa); |
| 1527 | return false; | 1517 | return false; |
| 1528 | }; | 1518 | }; |
| 1529 | defer lib_stub.deinit(); | 1519 | defer lib_stub.deinit(); |
| 1530 | | 1520 | |
| 1531 | try dylib.parseFromStub( | 1521 | try dylib.parseFromStub( |
| 1532 | self.base.allocator, | 1522 | gpa, |
| 1533 | self.base.options.target, | 1523 | self.base.options.target, |
| 1534 | lib_stub, | 1524 | lib_stub, |
| 1535 | dylib_id, | 1525 | dylib_id, |
| 1536 | dependent_libs, | 1526 | dependent_libs, |
| | 1527 | path, |
| 1537 | ); | 1528 | ); |
| 1538 | }, | 1529 | }, |
| 1539 | else => |e| return e, | 1530 | else => |e| return e, |
| ... | @@ -1547,13 +1538,13 @@ pub fn parseDylib( | ... | @@ -1547,13 +1538,13 @@ pub fn parseDylib( |
| 1547 | log.warn(" dylib version: {}", .{dylib.id.?.current_version}); | 1538 | log.warn(" dylib version: {}", .{dylib.id.?.current_version}); |
| 1548 | | 1539 | |
| 1549 | // TODO maybe this should be an error and facilitate auto-cleanup? | 1540 | // TODO maybe this should be an error and facilitate auto-cleanup? |
| 1550 | dylib.deinit(self.base.allocator); | 1541 | dylib.deinit(gpa); |
| 1551 | return false; | 1542 | return false; |
| 1552 | } | 1543 | } |
| 1553 | } | 1544 | } |
| 1554 | | 1545 | |
| 1555 | try self.dylibs.append(self.base.allocator, dylib); | 1546 | try self.dylibs.append(gpa, dylib); |
| 1556 | try self.dylibs_map.putNoClobber(self.base.allocator, dylib.id.?.name, dylib_id); | 1547 | try self.dylibs_map.putNoClobber(gpa, dylib.id.?.name, dylib_id); |
| 1557 | | 1548 | |
| 1558 | const should_link_dylib_even_if_unreachable = blk: { | 1549 | const should_link_dylib_even_if_unreachable = blk: { |
| 1559 | if (self.base.options.dead_strip_dylibs and !opts.needed) break :blk false; | 1550 | if (self.base.options.dead_strip_dylibs and !opts.needed) break :blk false; |
| ... | @@ -1561,8 +1552,7 @@ pub fn parseDylib( | ... | @@ -1561,8 +1552,7 @@ pub fn parseDylib( |
| 1561 | }; | 1552 | }; |
| 1562 | | 1553 | |
| 1563 | if (should_link_dylib_even_if_unreachable) { | 1554 | if (should_link_dylib_even_if_unreachable) { |
| 1564 | try self.addLoadDylibLC(dylib_id); | 1555 | try self.referenced_dylibs.putNoClobber(gpa, dylib_id, {}); |
| 1565 | try self.referenced_dylibs.putNoClobber(self.base.allocator, dylib_id, {}); | | |
| 1566 | } | 1556 | } |
| 1567 | | 1557 | |
| 1568 | return true; | 1558 | return true; |
| ... | @@ -1572,10 +1562,8 @@ fn parseInputFiles(self: *MachO, files: []const []const u8, syslibroot: ?[]const | ... | @@ -1572,10 +1562,8 @@ fn parseInputFiles(self: *MachO, files: []const []const u8, syslibroot: ?[]const |
| 1572 | for (files) |file_name| { | 1562 | for (files) |file_name| { |
| 1573 | const full_path = full_path: { | 1563 | const full_path = full_path: { |
| 1574 | var buffer: [fs.MAX_PATH_BYTES]u8 = undefined; | 1564 | var buffer: [fs.MAX_PATH_BYTES]u8 = undefined; |
| 1575 | const path = try fs.realpath(file_name, &buffer); | 1565 | break :full_path try fs.realpath(file_name, &buffer); |
| 1576 | break :full_path try self.base.allocator.dupe(u8, path); | | |
| 1577 | }; | 1566 | }; |
| 1578 | defer self.base.allocator.free(full_path); | | |
| 1579 | log.debug("parsing input file path '{s}'", .{full_path}); | 1567 | log.debug("parsing input file path '{s}'", .{full_path}); |
| 1580 | | 1568 | |
| 1581 | if (try self.parseObject(full_path)) continue; | 1569 | if (try self.parseObject(full_path)) continue; |
| ... | @@ -1592,10 +1580,8 @@ fn parseAndForceLoadStaticArchives(self: *MachO, files: []const []const u8) !voi | ... | @@ -1592,10 +1580,8 @@ fn parseAndForceLoadStaticArchives(self: *MachO, files: []const []const u8) !voi |
| 1592 | for (files) |file_name| { | 1580 | for (files) |file_name| { |
| 1593 | const full_path = full_path: { | 1581 | const full_path = full_path: { |
| 1594 | var buffer: [fs.MAX_PATH_BYTES]u8 = undefined; | 1582 | var buffer: [fs.MAX_PATH_BYTES]u8 = undefined; |
| 1595 | const path = try fs.realpath(file_name, &buffer); | 1583 | break :full_path try fs.realpath(file_name, &buffer); |
| 1596 | break :full_path try self.base.allocator.dupe(u8, path); | | |
| 1597 | }; | 1584 | }; |
| 1598 | defer self.base.allocator.free(full_path); | | |
| 1599 | log.debug("parsing and force loading static archive '{s}'", .{full_path}); | 1585 | log.debug("parsing and force loading static archive '{s}'", .{full_path}); |
| 1600 | | 1586 | |
| 1601 | if (try self.parseArchive(full_path, true)) continue; | 1587 | if (try self.parseArchive(full_path, true)) continue; |
| ... | @@ -1669,639 +1655,232 @@ fn parseDependentLibs(self: *MachO, syslibroot: ?[]const u8, dependent_libs: any | ... | @@ -1669,639 +1655,232 @@ fn parseDependentLibs(self: *MachO, syslibroot: ?[]const u8, dependent_libs: any |
| 1669 | } | 1655 | } |
| 1670 | } | 1656 | } |
| 1671 | | 1657 | |
| 1672 | pub const MatchingSection = struct { | 1658 | pub fn getOutputSection(self: *MachO, sect: macho.section_64) !?u8 { |
| 1673 | seg: u16, | | |
| 1674 | sect: u16, | | |
| 1675 | | | |
| 1676 | pub fn eql(this: MatchingSection, other: struct { | | |
| 1677 | seg: ?u16, | | |
| 1678 | sect: ?u16, | | |
| 1679 | }) bool { | | |
| 1680 | const seg = other.seg orelse return false; | | |
| 1681 | const sect = other.sect orelse return false; | | |
| 1682 | return this.seg == seg and this.sect == sect; | | |
| 1683 | } | | |
| 1684 | }; | | |
| 1685 | | | |
| 1686 | pub fn getMatchingSection(self: *MachO, sect: macho.section_64) !?MatchingSection { | | |
| 1687 | const segname = sect.segName(); | 1659 | const segname = sect.segName(); |
| 1688 | const sectname = sect.sectName(); | 1660 | const sectname = sect.sectName(); |
| 1689 | const res: ?MatchingSection = blk: { | 1661 | const res: ?u8 = blk: { |
| 1690 | switch (sect.type_()) { | 1662 | if (mem.eql(u8, "__LLVM", segname)) { |
| 1691 | macho.S_4BYTE_LITERALS, macho.S_8BYTE_LITERALS, macho.S_16BYTE_LITERALS => { | 1663 | log.debug("TODO LLVM section: type 0x{x}, name '{s},{s}'", .{ |
| 1692 | if (self.text_const_section_index == null) { | 1664 | sect.flags, segname, sectname, |
| 1693 | self.text_const_section_index = try self.initSection( | 1665 | }); |
| 1694 | self.text_segment_cmd_index.?, | 1666 | break :blk null; |
| 1695 | "__const", | 1667 | } |
| | 1668 | |
| | 1669 | if (sect.isCode()) { |
| | 1670 | if (self.text_section_index == null) { |
| | 1671 | self.text_section_index = try self.initSection( |
| | 1672 | "__TEXT", |
| | 1673 | "__text", |
| | 1674 | sect.size, |
| | 1675 | sect.@"align", |
| | 1676 | .{ |
| | 1677 | .flags = macho.S_REGULAR | |
| | 1678 | macho.S_ATTR_PURE_INSTRUCTIONS | |
| | 1679 | macho.S_ATTR_SOME_INSTRUCTIONS, |
| | 1680 | }, |
| | 1681 | ); |
| | 1682 | } |
| | 1683 | break :blk self.text_section_index.?; |
| | 1684 | } |
| | 1685 | |
| | 1686 | if (sect.isDebug()) { |
| | 1687 | // TODO debug attributes |
| | 1688 | if (mem.eql(u8, "__LD", segname) and mem.eql(u8, "__compact_unwind", sectname)) { |
| | 1689 | log.debug("TODO compact unwind section: type 0x{x}, name '{s},{s}'", .{ |
| | 1690 | sect.flags, segname, sectname, |
| | 1691 | }); |
| | 1692 | } |
| | 1693 | break :blk null; |
| | 1694 | } |
| | 1695 | |
| | 1696 | switch (sect.@"type"()) { |
| | 1697 | macho.S_4BYTE_LITERALS, |
| | 1698 | macho.S_8BYTE_LITERALS, |
| | 1699 | macho.S_16BYTE_LITERALS, |
| | 1700 | => { |
| | 1701 | break :blk self.getSectionByName("__TEXT", "__const") orelse try self.initSection( |
| | 1702 | "__TEXT", |
| | 1703 | "__const", |
| | 1704 | sect.size, |
| | 1705 | sect.@"align", |
| | 1706 | .{}, |
| | 1707 | ); |
| | 1708 | }, |
| | 1709 | macho.S_CSTRING_LITERALS => { |
| | 1710 | if (mem.startsWith(u8, sectname, "__objc")) { |
| | 1711 | break :blk self.getSectionByName(segname, sectname) orelse try self.initSection( |
| | 1712 | segname, |
| | 1713 | sectname, |
| 1696 | sect.size, | 1714 | sect.size, |
| 1697 | sect.@"align", | 1715 | sect.@"align", |
| 1698 | .{}, | 1716 | .{}, |
| 1699 | ); | 1717 | ); |
| 1700 | } | 1718 | } |
| 1701 | | 1719 | break :blk self.getSectionByName("__TEXT", "__cstring") orelse try self.initSection( |
| 1702 | break :blk .{ | 1720 | "__TEXT", |
| 1703 | .seg = self.text_segment_cmd_index.?, | 1721 | "__cstring", |
| 1704 | .sect = self.text_const_section_index.?, | 1722 | sect.size, |
| 1705 | }; | 1723 | sect.@"align", |
| | 1724 | .{ .flags = macho.S_CSTRING_LITERALS }, |
| | 1725 | ); |
| 1706 | }, | 1726 | }, |
| 1707 | macho.S_CSTRING_LITERALS => { | 1727 | macho.S_MOD_INIT_FUNC_POINTERS, |
| 1708 | if (mem.eql(u8, sectname, "__objc_methname")) { | 1728 | macho.S_MOD_TERM_FUNC_POINTERS, |
| 1709 | // TODO it seems the common values within the sections in objects are deduplicated/merged | 1729 | => { |
| 1710 | // on merging the sections' contents. | 1730 | break :blk self.getSectionByName("__DATA_CONST", sectname) orelse try self.initSection( |
| 1711 | if (self.objc_methname_section_index == null) { | 1731 | "__DATA_CONST", |
| 1712 | self.objc_methname_section_index = try self.initSection( | 1732 | sectname, |
| 1713 | self.text_segment_cmd_index.?, | 1733 | sect.size, |
| 1714 | "__objc_methname", | 1734 | sect.@"align", |
| | 1735 | .{ .flags = sect.flags }, |
| | 1736 | ); |
| | 1737 | }, |
| | 1738 | macho.S_LITERAL_POINTERS, |
| | 1739 | macho.S_ZEROFILL, |
| | 1740 | macho.S_THREAD_LOCAL_VARIABLES, |
| | 1741 | macho.S_THREAD_LOCAL_VARIABLE_POINTERS, |
| | 1742 | macho.S_THREAD_LOCAL_REGULAR, |
| | 1743 | macho.S_THREAD_LOCAL_ZEROFILL, |
| | 1744 | => { |
| | 1745 | break :blk self.getSectionByName(segname, sectname) orelse try self.initSection( |
| | 1746 | segname, |
| | 1747 | sectname, |
| | 1748 | sect.size, |
| | 1749 | sect.@"align", |
| | 1750 | .{ .flags = sect.flags }, |
| | 1751 | ); |
| | 1752 | }, |
| | 1753 | macho.S_COALESCED => { |
| | 1754 | break :blk self.getSectionByName(segname, sectname) orelse try self.initSection( |
| | 1755 | segname, |
| | 1756 | sectname, |
| | 1757 | sect.size, |
| | 1758 | sect.@"align", |
| | 1759 | .{}, |
| | 1760 | ); |
| | 1761 | }, |
| | 1762 | macho.S_REGULAR => { |
| | 1763 | if (mem.eql(u8, segname, "__TEXT")) { |
| | 1764 | if (mem.eql(u8, sectname, "__rodata") or |
| | 1765 | mem.eql(u8, sectname, "__typelink") or |
| | 1766 | mem.eql(u8, sectname, "__itablink") or |
| | 1767 | mem.eql(u8, sectname, "__gosymtab") or |
| | 1768 | mem.eql(u8, sectname, "__gopclntab")) |
| | 1769 | { |
| | 1770 | break :blk self.getSectionByName("__DATA_CONST", "__const") orelse try self.initSection( |
| | 1771 | "__DATA_CONST", |
| | 1772 | "__const", |
| 1715 | sect.size, | 1773 | sect.size, |
| 1716 | sect.@"align", | 1774 | sect.@"align", |
| 1717 | .{}, | 1775 | .{}, |
| 1718 | ); | 1776 | ); |
| 1719 | } | 1777 | } |
| 1720 | | 1778 | } |
| 1721 | break :blk .{ | 1779 | if (mem.eql(u8, segname, "__DATA")) { |
| 1722 | .seg = self.text_segment_cmd_index.?, | 1780 | if (mem.eql(u8, sectname, "__const") or |
| 1723 | .sect = self.objc_methname_section_index.?, | 1781 | mem.eql(u8, sectname, "__cfstring") or |
| 1724 | }; | 1782 | mem.eql(u8, sectname, "__objc_classlist") or |
| 1725 | } else if (mem.eql(u8, sectname, "__objc_methtype")) { | 1783 | mem.eql(u8, sectname, "__objc_imageinfo")) |
| 1726 | if (self.objc_methtype_section_index == null) { | 1784 | { |
| 1727 | self.objc_methtype_section_index = try self.initSection( | 1785 | break :blk self.getSectionByName("__DATA_CONST", sectname) orelse |
| 1728 | self.text_segment_cmd_index.?, | 1786 | try self.initSection( |
| 1729 | "__objc_methtype", | 1787 | "__DATA_CONST", |
| | 1788 | sectname, |
| 1730 | sect.size, | 1789 | sect.size, |
| 1731 | sect.@"align", | 1790 | sect.@"align", |
| 1732 | .{}, | 1791 | .{}, |
| 1733 | ); | 1792 | ); |
| | 1793 | } else if (mem.eql(u8, sectname, "__data")) { |
| | 1794 | if (self.data_section_index == null) { |
| | 1795 | self.data_section_index = try self.initSection( |
| | 1796 | segname, |
| | 1797 | sectname, |
| | 1798 | sect.size, |
| | 1799 | sect.@"align", |
| | 1800 | .{}, |
| | 1801 | ); |
| | 1802 | } |
| | 1803 | break :blk self.data_section_index.?; |
| 1734 | } | 1804 | } |
| | 1805 | } |
| | 1806 | break :blk self.getSectionByName(segname, sectname) orelse try self.initSection( |
| | 1807 | segname, |
| | 1808 | sectname, |
| | 1809 | sect.size, |
| | 1810 | sect.@"align", |
| | 1811 | .{}, |
| | 1812 | ); |
| | 1813 | }, |
| | 1814 | else => break :blk null, |
| | 1815 | } |
| | 1816 | }; |
| | 1817 | return res; |
| | 1818 | } |
| 1735 | | 1819 | |
| 1736 | break :blk .{ | 1820 | pub fn createEmptyAtom(gpa: Allocator, sym_index: u32, size: u64, alignment: u32) !*Atom { |
| 1737 | .seg = self.text_segment_cmd_index.?, | 1821 | const size_usize = math.cast(usize, size) orelse return error.Overflow; |
| 1738 | .sect = self.objc_methtype_section_index.?, | 1822 | const atom = try gpa.create(Atom); |
| 1739 | }; | 1823 | errdefer gpa.destroy(atom); |
| 1740 | } else if (mem.eql(u8, sectname, "__objc_classname")) { | 1824 | atom.* = Atom.empty; |
| 1741 | if (self.objc_classname_section_index == null) { | 1825 | atom.sym_index = sym_index; |
| 1742 | self.objc_classname_section_index = try self.initSection( | 1826 | atom.size = size; |
| 1743 | self.text_segment_cmd_index.?, | 1827 | atom.alignment = alignment; |
| 1744 | "__objc_classname", | | |
| 1745 | sect.size, | | |
| 1746 | sect.@"align", | | |
| 1747 | .{}, | | |
| 1748 | ); | | |
| 1749 | } | | |
| 1750 | | 1828 | |
| 1751 | break :blk .{ | 1829 | try atom.code.resize(gpa, size_usize); |
| 1752 | .seg = self.text_segment_cmd_index.?, | 1830 | mem.set(u8, atom.code.items, 0); |
| 1753 | .sect = self.objc_classname_section_index.?, | | |
| 1754 | }; | | |
| 1755 | } | | |
| 1756 | | 1831 | |
| 1757 | if (self.cstring_section_index == null) { | 1832 | return atom; |
| 1758 | self.cstring_section_index = try self.initSection( | 1833 | } |
| 1759 | self.text_segment_cmd_index.?, | | |
| 1760 | "__cstring", | | |
| 1761 | sect.size, | | |
| 1762 | sect.@"align", | | |
| 1763 | .{ | | |
| 1764 | .flags = macho.S_CSTRING_LITERALS, | | |
| 1765 | }, | | |
| 1766 | ); | | |
| 1767 | } | | |
| 1768 | | 1834 | |
| 1769 | break :blk .{ | 1835 | pub fn writeAtom(self: *MachO, atom: *Atom, sect_id: u8) !void { |
| 1770 | .seg = self.text_segment_cmd_index.?, | 1836 | const section = self.sections.get(sect_id); |
| 1771 | .sect = self.cstring_section_index.?, | 1837 | const sym = atom.getSymbol(self); |
| 1772 | }; | 1838 | const file_offset = section.header.offset + sym.n_value - section.header.addr; |
| 1773 | }, | 1839 | try atom.resolveRelocs(self); |
| 1774 | macho.S_LITERAL_POINTERS => { | 1840 | log.debug("writing atom for symbol {s} at file offset 0x{x}", .{ atom.getName(self), file_offset }); |
| 1775 | if (mem.eql(u8, segname, "__DATA") and mem.eql(u8, sectname, "__objc_selrefs")) { | 1841 | try self.base.file.?.pwriteAll(atom.code.items, file_offset); |
| 1776 | if (self.objc_selrefs_section_index == null) { | 1842 | } |
| 1777 | self.objc_selrefs_section_index = try self.initSection( | | |
| 1778 | self.data_segment_cmd_index.?, | | |
| 1779 | "__objc_selrefs", | | |
| 1780 | sect.size, | | |
| 1781 | sect.@"align", | | |
| 1782 | .{ | | |
| 1783 | .flags = macho.S_LITERAL_POINTERS, | | |
| 1784 | }, | | |
| 1785 | ); | | |
| 1786 | } | | |
| 1787 | | 1843 | |
| 1788 | break :blk .{ | 1844 | fn allocateSymbols(self: *MachO) !void { |
| 1789 | .seg = self.data_segment_cmd_index.?, | 1845 | const slice = self.sections.slice(); |
| 1790 | .sect = self.objc_selrefs_section_index.?, | 1846 | for (slice.items(.last_atom)) |last_atom, sect_id| { |
| 1791 | }; | 1847 | const header = slice.items(.header)[sect_id]; |
| 1792 | } else { | 1848 | var atom = last_atom orelse continue; |
| 1793 | // TODO investigate | | |
| 1794 | break :blk null; | | |
| 1795 | } | | |
| 1796 | }, | | |
| 1797 | macho.S_MOD_INIT_FUNC_POINTERS => { | | |
| 1798 | if (self.mod_init_func_section_index == null) { | | |
| 1799 | self.mod_init_func_section_index = try self.initSection( | | |
| 1800 | self.data_const_segment_cmd_index.?, | | |
| 1801 | "__mod_init_func", | | |
| 1802 | sect.size, | | |
| 1803 | sect.@"align", | | |
| 1804 | .{ | | |
| 1805 | .flags = macho.S_MOD_INIT_FUNC_POINTERS, | | |
| 1806 | }, | | |
| 1807 | ); | | |
| 1808 | } | | |
| 1809 | | 1849 | |
| 1810 | break :blk .{ | 1850 | while (atom.prev) |prev| { |
| 1811 | .seg = self.data_const_segment_cmd_index.?, | 1851 | atom = prev; |
| 1812 | .sect = self.mod_init_func_section_index.?, | 1852 | } |
| 1813 | }; | | |
| 1814 | }, | | |
| 1815 | macho.S_MOD_TERM_FUNC_POINTERS => { | | |
| 1816 | if (self.mod_term_func_section_index == null) { | | |
| 1817 | self.mod_term_func_section_index = try self.initSection( | | |
| 1818 | self.data_const_segment_cmd_index.?, | | |
| 1819 | "__mod_term_func", | | |
| 1820 | sect.size, | | |
| 1821 | sect.@"align", | | |
| 1822 | .{ | | |
| 1823 | .flags = macho.S_MOD_TERM_FUNC_POINTERS, | | |
| 1824 | }, | | |
| 1825 | ); | | |
| 1826 | } | | |
| 1827 | | 1853 | |
| 1828 | break :blk .{ | 1854 | const n_sect = @intCast(u8, sect_id + 1); |
| 1829 | .seg = self.data_const_segment_cmd_index.?, | 1855 | var base_vaddr = header.addr; |
| 1830 | .sect = self.mod_term_func_section_index.?, | | |
| 1831 | }; | | |
| 1832 | }, | | |
| 1833 | macho.S_ZEROFILL => { | | |
| 1834 | if (self.bss_section_index == null) { | | |
| 1835 | self.bss_section_index = try self.initSection( | | |
| 1836 | self.data_segment_cmd_index.?, | | |
| 1837 | "__bss", | | |
| 1838 | sect.size, | | |
| 1839 | sect.@"align", | | |
| 1840 | .{ | | |
| 1841 | .flags = macho.S_ZEROFILL, | | |
| 1842 | }, | | |
| 1843 | ); | | |
| 1844 | } | | |
| 1845 | | 1856 | |
| 1846 | break :blk .{ | 1857 | log.debug("allocating local symbols in sect({d}, '{s},{s}')", .{ |
| 1847 | .seg = self.data_segment_cmd_index.?, | 1858 | n_sect, |
| 1848 | .sect = self.bss_section_index.?, | 1859 | header.segName(), |
| 1849 | }; | 1860 | header.sectName(), |
| 1850 | }, | 1861 | }); |
| 1851 | macho.S_THREAD_LOCAL_VARIABLES => { | | |
| 1852 | if (self.tlv_section_index == null) { | | |
| 1853 | self.tlv_section_index = try self.initSection( | | |
| 1854 | self.data_segment_cmd_index.?, | | |
| 1855 | "__thread_vars", | | |
| 1856 | sect.size, | | |
| 1857 | sect.@"align", | | |
| 1858 | .{ | | |
| 1859 | .flags = macho.S_THREAD_LOCAL_VARIABLES, | | |
| 1860 | }, | | |
| 1861 | ); | | |
| 1862 | } | | |
| 1863 | | 1862 | |
| 1864 | break :blk .{ | 1863 | while (true) { |
| 1865 | .seg = self.data_segment_cmd_index.?, | 1864 | const alignment = try math.powi(u32, 2, atom.alignment); |
| 1866 | .sect = self.tlv_section_index.?, | 1865 | base_vaddr = mem.alignForwardGeneric(u64, base_vaddr, alignment); |
| 1867 | }; | | |
| 1868 | }, | | |
| 1869 | macho.S_THREAD_LOCAL_VARIABLE_POINTERS => { | | |
| 1870 | if (self.tlv_ptrs_section_index == null) { | | |
| 1871 | self.tlv_ptrs_section_index = try self.initSection( | | |
| 1872 | self.data_segment_cmd_index.?, | | |
| 1873 | "__thread_ptrs", | | |
| 1874 | sect.size, | | |
| 1875 | sect.@"align", | | |
| 1876 | .{ | | |
| 1877 | .flags = macho.S_THREAD_LOCAL_VARIABLE_POINTERS, | | |
| 1878 | }, | | |
| 1879 | ); | | |
| 1880 | } | | |
| 1881 | | 1866 | |
| 1882 | break :blk .{ | 1867 | const sym = atom.getSymbolPtr(self); |
| 1883 | .seg = self.data_segment_cmd_index.?, | 1868 | sym.n_value = base_vaddr; |
| 1884 | .sect = self.tlv_ptrs_section_index.?, | 1869 | sym.n_sect = n_sect; |
| 1885 | }; | | |
| 1886 | }, | | |
| 1887 | macho.S_THREAD_LOCAL_REGULAR => { | | |
| 1888 | if (self.tlv_data_section_index == null) { | | |
| 1889 | self.tlv_data_section_index = try self.initSection( | | |
| 1890 | self.data_segment_cmd_index.?, | | |
| 1891 | "__thread_data", | | |
| 1892 | sect.size, | | |
| 1893 | sect.@"align", | | |
| 1894 | .{ | | |
| 1895 | .flags = macho.S_THREAD_LOCAL_REGULAR, | | |
| 1896 | }, | | |
| 1897 | ); | | |
| 1898 | } | | |
| 1899 | | 1870 | |
| 1900 | break :blk .{ | 1871 | log.debug(" ATOM(%{d}, '{s}') @{x}", .{ atom.sym_index, atom.getName(self), base_vaddr }); |
| 1901 | .seg = self.data_segment_cmd_index.?, | | |
| 1902 | .sect = self.tlv_data_section_index.?, | | |
| 1903 | }; | | |
| 1904 | }, | | |
| 1905 | macho.S_THREAD_LOCAL_ZEROFILL => { | | |
| 1906 | if (self.tlv_bss_section_index == null) { | | |
| 1907 | self.tlv_bss_section_index = try self.initSection( | | |
| 1908 | self.data_segment_cmd_index.?, | | |
| 1909 | "__thread_bss", | | |
| 1910 | sect.size, | | |
| 1911 | sect.@"align", | | |
| 1912 | .{ | | |
| 1913 | .flags = macho.S_THREAD_LOCAL_ZEROFILL, | | |
| 1914 | }, | | |
| 1915 | ); | | |
| 1916 | } | | |
| 1917 | | 1872 | |
| 1918 | break :blk .{ | 1873 | // Update each symbol contained within the atom |
| 1919 | .seg = self.data_segment_cmd_index.?, | 1874 | for (atom.contained.items) |sym_at_off| { |
| 1920 | .sect = self.tlv_bss_section_index.?, | 1875 | const contained_sym = self.getSymbolPtr(.{ |
| 1921 | }; | 1876 | .sym_index = sym_at_off.sym_index, |
| 1922 | }, | 1877 | .file = atom.file, |
| 1923 | macho.S_COALESCED => { | 1878 | }); |
| 1924 | if (mem.eql(u8, "__TEXT", segname) and mem.eql(u8, "__eh_frame", sectname)) { | 1879 | contained_sym.n_value = base_vaddr + sym_at_off.offset; |
| 1925 | // TODO I believe __eh_frame is currently part of __unwind_info section | 1880 | contained_sym.n_sect = n_sect; |
| 1926 | // in the latest ld64 output. | 1881 | } |
| 1927 | if (self.eh_frame_section_index == null) { | | |
| 1928 | self.eh_frame_section_index = try self.initSection( | | |
| 1929 | self.text_segment_cmd_index.?, | | |
| 1930 | "__eh_frame", | | |
| 1931 | sect.size, | | |
| 1932 | sect.@"align", | | |
| 1933 | .{}, | | |
| 1934 | ); | | |
| 1935 | } | | |
| 1936 | | 1882 | |
| 1937 | break :blk .{ | 1883 | base_vaddr += atom.size; |
| 1938 | .seg = self.text_segment_cmd_index.?, | | |
| 1939 | .sect = self.eh_frame_section_index.?, | | |
| 1940 | }; | | |
| 1941 | } | | |
| 1942 | | | |
| 1943 | // TODO audit this: is this the right mapping? | | |
| 1944 | if (self.data_const_section_index == null) { | | |
| 1945 | self.data_const_section_index = try self.initSection( | | |
| 1946 | self.data_const_segment_cmd_index.?, | | |
| 1947 | "__const", | | |
| 1948 | sect.size, | | |
| 1949 | sect.@"align", | | |
| 1950 | .{}, | | |
| 1951 | ); | | |
| 1952 | } | | |
| 1953 | | | |
| 1954 | break :blk .{ | | |
| 1955 | .seg = self.data_const_segment_cmd_index.?, | | |
| 1956 | .sect = self.data_const_section_index.?, | | |
| 1957 | }; | | |
| 1958 | }, | | |
| 1959 | macho.S_REGULAR => { | | |
| 1960 | if (sect.isCode()) { | | |
| 1961 | if (self.text_section_index == null) { | | |
| 1962 | self.text_section_index = try self.initSection( | | |
| 1963 | self.text_segment_cmd_index.?, | | |
| 1964 | "__text", | | |
| 1965 | sect.size, | | |
| 1966 | sect.@"align", | | |
| 1967 | .{ | | |
| 1968 | .flags = macho.S_REGULAR | | | |
| 1969 | macho.S_ATTR_PURE_INSTRUCTIONS | | | |
| 1970 | macho.S_ATTR_SOME_INSTRUCTIONS, | | |
| 1971 | }, | | |
| 1972 | ); | | |
| 1973 | } | | |
| 1974 | | | |
| 1975 | break :blk .{ | | |
| 1976 | .seg = self.text_segment_cmd_index.?, | | |
| 1977 | .sect = self.text_section_index.?, | | |
| 1978 | }; | | |
| 1979 | } | | |
| 1980 | if (sect.isDebug()) { | | |
| 1981 | // TODO debug attributes | | |
| 1982 | if (mem.eql(u8, "__LD", segname) and mem.eql(u8, "__compact_unwind", sectname)) { | | |
| 1983 | log.debug("TODO compact unwind section: type 0x{x}, name '{s},{s}'", .{ | | |
| 1984 | sect.flags, segname, sectname, | | |
| 1985 | }); | | |
| 1986 | } | | |
| 1987 | break :blk null; | | |
| 1988 | } | | |
| 1989 | | | |
| 1990 | if (mem.eql(u8, segname, "__TEXT")) { | | |
| 1991 | if (mem.eql(u8, sectname, "__ustring")) { | | |
| 1992 | if (self.ustring_section_index == null) { | | |
| 1993 | self.ustring_section_index = try self.initSection( | | |
| 1994 | self.text_segment_cmd_index.?, | | |
| 1995 | "__ustring", | | |
| 1996 | sect.size, | | |
| 1997 | sect.@"align", | | |
| 1998 | .{}, | | |
| 1999 | ); | | |
| 2000 | } | | |
| 2001 | | | |
| 2002 | break :blk .{ | | |
| 2003 | .seg = self.text_segment_cmd_index.?, | | |
| 2004 | .sect = self.ustring_section_index.?, | | |
| 2005 | }; | | |
| 2006 | } else if (mem.eql(u8, sectname, "__gcc_except_tab")) { | | |
| 2007 | if (self.gcc_except_tab_section_index == null) { | | |
| 2008 | self.gcc_except_tab_section_index = try self.initSection( | | |
| 2009 | self.text_segment_cmd_index.?, | | |
| 2010 | "__gcc_except_tab", | | |
| 2011 | sect.size, | | |
| 2012 | sect.@"align", | | |
| 2013 | .{}, | | |
| 2014 | ); | | |
| 2015 | } | | |
| 2016 | | | |
| 2017 | break :blk .{ | | |
| 2018 | .seg = self.text_segment_cmd_index.?, | | |
| 2019 | .sect = self.gcc_except_tab_section_index.?, | | |
| 2020 | }; | | |
| 2021 | } else if (mem.eql(u8, sectname, "__objc_methlist")) { | | |
| 2022 | if (self.objc_methlist_section_index == null) { | | |
| 2023 | self.objc_methlist_section_index = try self.initSection( | | |
| 2024 | self.text_segment_cmd_index.?, | | |
| 2025 | "__objc_methlist", | | |
| 2026 | sect.size, | | |
| 2027 | sect.@"align", | | |
| 2028 | .{}, | | |
| 2029 | ); | | |
| 2030 | } | | |
| 2031 | | | |
| 2032 | break :blk .{ | | |
| 2033 | .seg = self.text_segment_cmd_index.?, | | |
| 2034 | .sect = self.objc_methlist_section_index.?, | | |
| 2035 | }; | | |
| 2036 | } else if (mem.eql(u8, sectname, "__rodata") or | | |
| 2037 | mem.eql(u8, sectname, "__typelink") or | | |
| 2038 | mem.eql(u8, sectname, "__itablink") or | | |
| 2039 | mem.eql(u8, sectname, "__gosymtab") or | | |
| 2040 | mem.eql(u8, sectname, "__gopclntab")) | | |
| 2041 | { | | |
| 2042 | if (self.data_const_section_index == null) { | | |
| 2043 | self.data_const_section_index = try self.initSection( | | |
| 2044 | self.data_const_segment_cmd_index.?, | | |
| 2045 | "__const", | | |
| 2046 | sect.size, | | |
| 2047 | sect.@"align", | | |
| 2048 | .{}, | | |
| 2049 | ); | | |
| 2050 | } | | |
| 2051 | | | |
| 2052 | break :blk .{ | | |
| 2053 | .seg = self.data_const_segment_cmd_index.?, | | |
| 2054 | .sect = self.data_const_section_index.?, | | |
| 2055 | }; | | |
| 2056 | } else { | | |
| 2057 | if (self.text_const_section_index == null) { | | |
| 2058 | self.text_const_section_index = try self.initSection( | | |
| 2059 | self.text_segment_cmd_index.?, | | |
| 2060 | "__const", | | |
| 2061 | sect.size, | | |
| 2062 | sect.@"align", | | |
| 2063 | .{}, | | |
| 2064 | ); | | |
| 2065 | } | | |
| 2066 | | | |
| 2067 | break :blk .{ | | |
| 2068 | .seg = self.text_segment_cmd_index.?, | | |
| 2069 | .sect = self.text_const_section_index.?, | | |
| 2070 | }; | | |
| 2071 | } | | |
| 2072 | } | | |
| 2073 | | | |
| 2074 | if (mem.eql(u8, segname, "__DATA_CONST")) { | | |
| 2075 | if (self.data_const_section_index == null) { | | |
| 2076 | self.data_const_section_index = try self.initSection( | | |
| 2077 | self.data_const_segment_cmd_index.?, | | |
| 2078 | "__const", | | |
| 2079 | sect.size, | | |
| 2080 | sect.@"align", | | |
| 2081 | .{}, | | |
| 2082 | ); | | |
| 2083 | } | | |
| 2084 | | | |
| 2085 | break :blk .{ | | |
| 2086 | .seg = self.data_const_segment_cmd_index.?, | | |
| 2087 | .sect = self.data_const_section_index.?, | | |
| 2088 | }; | | |
| 2089 | } | | |
| 2090 | | | |
| 2091 | if (mem.eql(u8, segname, "__DATA")) { | | |
| 2092 | if (mem.eql(u8, sectname, "__const")) { | | |
| 2093 | if (self.data_const_section_index == null) { | | |
| 2094 | self.data_const_section_index = try self.initSection( | | |
| 2095 | self.data_const_segment_cmd_index.?, | | |
| 2096 | "__const", | | |
| 2097 | sect.size, | | |
| 2098 | sect.@"align", | | |
| 2099 | .{}, | | |
| 2100 | ); | | |
| 2101 | } | | |
| 2102 | | | |
| 2103 | break :blk .{ | | |
| 2104 | .seg = self.data_const_segment_cmd_index.?, | | |
| 2105 | .sect = self.data_const_section_index.?, | | |
| 2106 | }; | | |
| 2107 | } else if (mem.eql(u8, sectname, "__cfstring")) { | | |
| 2108 | if (self.objc_cfstring_section_index == null) { | | |
| 2109 | self.objc_cfstring_section_index = try self.initSection( | | |
| 2110 | self.data_const_segment_cmd_index.?, | | |
| 2111 | "__cfstring", | | |
| 2112 | sect.size, | | |
| 2113 | sect.@"align", | | |
| 2114 | .{}, | | |
| 2115 | ); | | |
| 2116 | } | | |
| 2117 | | | |
| 2118 | break :blk .{ | | |
| 2119 | .seg = self.data_const_segment_cmd_index.?, | | |
| 2120 | .sect = self.objc_cfstring_section_index.?, | | |
| 2121 | }; | | |
| 2122 | } else if (mem.eql(u8, sectname, "__objc_classlist")) { | | |
| 2123 | if (self.objc_classlist_section_index == null) { | | |
| 2124 | self.objc_classlist_section_index = try self.initSection( | | |
| 2125 | self.data_const_segment_cmd_index.?, | | |
| 2126 | "__objc_classlist", | | |
| 2127 | sect.size, | | |
| 2128 | sect.@"align", | | |
| 2129 | .{}, | | |
| 2130 | ); | | |
| 2131 | } | | |
| 2132 | | | |
| 2133 | break :blk .{ | | |
| 2134 | .seg = self.data_const_segment_cmd_index.?, | | |
| 2135 | .sect = self.objc_classlist_section_index.?, | | |
| 2136 | }; | | |
| 2137 | } else if (mem.eql(u8, sectname, "__objc_imageinfo")) { | | |
| 2138 | if (self.objc_imageinfo_section_index == null) { | | |
| 2139 | self.objc_imageinfo_section_index = try self.initSection( | | |
| 2140 | self.data_const_segment_cmd_index.?, | | |
| 2141 | "__objc_imageinfo", | | |
| 2142 | sect.size, | | |
| 2143 | sect.@"align", | | |
| 2144 | .{}, | | |
| 2145 | ); | | |
| 2146 | } | | |
| 2147 | | | |
| 2148 | break :blk .{ | | |
| 2149 | .seg = self.data_const_segment_cmd_index.?, | | |
| 2150 | .sect = self.objc_imageinfo_section_index.?, | | |
| 2151 | }; | | |
| 2152 | } else if (mem.eql(u8, sectname, "__objc_const")) { | | |
| 2153 | if (self.objc_const_section_index == null) { | | |
| 2154 | self.objc_const_section_index = try self.initSection( | | |
| 2155 | self.data_segment_cmd_index.?, | | |
| 2156 | "__objc_const", | | |
| 2157 | sect.size, | | |
| 2158 | sect.@"align", | | |
| 2159 | .{}, | | |
| 2160 | ); | | |
| 2161 | } | | |
| 2162 | | | |
| 2163 | break :blk .{ | | |
| 2164 | .seg = self.data_segment_cmd_index.?, | | |
| 2165 | .sect = self.objc_const_section_index.?, | | |
| 2166 | }; | | |
| 2167 | } else if (mem.eql(u8, sectname, "__objc_classrefs")) { | | |
| 2168 | if (self.objc_classrefs_section_index == null) { | | |
| 2169 | self.objc_classrefs_section_index = try self.initSection( | | |
| 2170 | self.data_segment_cmd_index.?, | | |
| 2171 | "__objc_classrefs", | | |
| 2172 | sect.size, | | |
| 2173 | sect.@"align", | | |
| 2174 | .{}, | | |
| 2175 | ); | | |
| 2176 | } | | |
| 2177 | | | |
| 2178 | break :blk .{ | | |
| 2179 | .seg = self.data_segment_cmd_index.?, | | |
| 2180 | .sect = self.objc_classrefs_section_index.?, | | |
| 2181 | }; | | |
| 2182 | } else if (mem.eql(u8, sectname, "__objc_data")) { | | |
| 2183 | if (self.objc_data_section_index == null) { | | |
| 2184 | self.objc_data_section_index = try self.initSection( | | |
| 2185 | self.data_segment_cmd_index.?, | | |
| 2186 | "__objc_data", | | |
| 2187 | sect.size, | | |
| 2188 | sect.@"align", | | |
| 2189 | .{}, | | |
| 2190 | ); | | |
| 2191 | } | | |
| 2192 | | | |
| 2193 | break :blk .{ | | |
| 2194 | .seg = self.data_segment_cmd_index.?, | | |
| 2195 | .sect = self.objc_data_section_index.?, | | |
| 2196 | }; | | |
| 2197 | } else if (mem.eql(u8, sectname, ".rustc")) { | | |
| 2198 | if (self.rustc_section_index == null) { | | |
| 2199 | self.rustc_section_index = try self.initSection( | | |
| 2200 | self.data_segment_cmd_index.?, | | |
| 2201 | ".rustc", | | |
| 2202 | sect.size, | | |
| 2203 | sect.@"align", | | |
| 2204 | .{}, | | |
| 2205 | ); | | |
| 2206 | // We need to preserve the section size for rustc to properly | | |
| 2207 | // decompress the metadata. | | |
| 2208 | self.rustc_section_size = sect.size; | | |
| 2209 | } | | |
| 2210 | | | |
| 2211 | break :blk .{ | | |
| 2212 | .seg = self.data_segment_cmd_index.?, | | |
| 2213 | .sect = self.rustc_section_index.?, | | |
| 2214 | }; | | |
| 2215 | } else { | | |
| 2216 | if (self.data_section_index == null) { | | |
| 2217 | self.data_section_index = try self.initSection( | | |
| 2218 | self.data_segment_cmd_index.?, | | |
| 2219 | "__data", | | |
| 2220 | sect.size, | | |
| 2221 | sect.@"align", | | |
| 2222 | .{}, | | |
| 2223 | ); | | |
| 2224 | } | | |
| 2225 | | | |
| 2226 | break :blk .{ | | |
| 2227 | .seg = self.data_segment_cmd_index.?, | | |
| 2228 | .sect = self.data_section_index.?, | | |
| 2229 | }; | | |
| 2230 | } | | |
| 2231 | } | | |
| 2232 | | | |
| 2233 | if (mem.eql(u8, "__LLVM", segname) and mem.eql(u8, "__asm", sectname)) { | | |
| 2234 | log.debug("TODO LLVM asm section: type 0x{x}, name '{s},{s}'", .{ | | |
| 2235 | sect.flags, segname, sectname, | | |
| 2236 | }); | | |
| 2237 | } | | |
| 2238 | | | |
| 2239 | break :blk null; | | |
| 2240 | }, | | |
| 2241 | else => break :blk null, | | |
| 2242 | } | | |
| 2243 | }; | | |
| 2244 | return res; | | |
| 2245 | } | | |
| 2246 | | | |
| 2247 | pub fn createEmptyAtom(gpa: Allocator, sym_index: u32, size: u64, alignment: u32) !*Atom { | | |
| 2248 | const size_usize = math.cast(usize, size) orelse return error.Overflow; | | |
| 2249 | const atom = try gpa.create(Atom); | | |
| 2250 | errdefer gpa.destroy(atom); | | |
| 2251 | atom.* = Atom.empty; | | |
| 2252 | atom.sym_index = sym_index; | | |
| 2253 | atom.size = size; | | |
| 2254 | atom.alignment = alignment; | | |
| 2255 | | | |
| 2256 | try atom.code.resize(gpa, size_usize); | | |
| 2257 | mem.set(u8, atom.code.items, 0); | | |
| 2258 | | | |
| 2259 | return atom; | | |
| 2260 | } | | |
| 2261 | | | |
| 2262 | pub fn writeAtom(self: *MachO, atom: *Atom, match: MatchingSection) !void { | | |
| 2263 | const sect = self.getSection(match); | | |
| 2264 | const sym = atom.getSymbol(self); | | |
| 2265 | const file_offset = sect.offset + sym.n_value - sect.addr; | | |
| 2266 | try atom.resolveRelocs(self); | | |
| 2267 | log.debug("writing atom for symbol {s} at file offset 0x{x}", .{ atom.getName(self), file_offset }); | | |
| 2268 | try self.base.file.?.pwriteAll(atom.code.items, file_offset); | | |
| 2269 | } | | |
| 2270 | | | |
| 2271 | fn allocateSymbols(self: *MachO) !void { | | |
| 2272 | var it = self.atoms.iterator(); | | |
| 2273 | while (it.next()) |entry| { | | |
| 2274 | const match = entry.key_ptr.*; | | |
| 2275 | var atom = entry.value_ptr.*; | | |
| 2276 | | | |
| 2277 | while (atom.prev) |prev| { | | |
| 2278 | atom = prev; | | |
| 2279 | } | | |
| 2280 | | | |
| 2281 | const n_sect = self.getSectionOrdinal(match); | | |
| 2282 | const sect = self.getSection(match); | | |
| 2283 | var base_vaddr = sect.addr; | | |
| 2284 | | | |
| 2285 | log.debug("allocating local symbols in sect({d}, '{s},{s}')", .{ n_sect, sect.segName(), sect.sectName() }); | | |
| 2286 | | | |
| 2287 | while (true) { | | |
| 2288 | const alignment = try math.powi(u32, 2, atom.alignment); | | |
| 2289 | base_vaddr = mem.alignForwardGeneric(u64, base_vaddr, alignment); | | |
| 2290 | | | |
| 2291 | const sym = atom.getSymbolPtr(self); | | |
| 2292 | sym.n_value = base_vaddr; | | |
| 2293 | sym.n_sect = n_sect; | | |
| 2294 | | | |
| 2295 | log.debug(" ATOM(%{d}, '{s}') @{x}", .{ atom.sym_index, atom.getName(self), base_vaddr }); | | |
| 2296 | | | |
| 2297 | // Update each symbol contained within the atom | | |
| 2298 | for (atom.contained.items) |sym_at_off| { | | |
| 2299 | const contained_sym = self.getSymbolPtr(.{ .sym_index = sym_at_off.sym_index, .file = atom.file }); | | |
| 2300 | contained_sym.n_value = base_vaddr + sym_at_off.offset; | | |
| 2301 | contained_sym.n_sect = n_sect; | | |
| 2302 | } | | |
| 2303 | | | |
| 2304 | base_vaddr += atom.size; | | |
| 2305 | | 1884 | |
| 2306 | if (atom.next) |next| { | 1885 | if (atom.next) |next| { |
| 2307 | atom = next; | 1886 | atom = next; |
| ... | @@ -2310,24 +1889,6 @@ fn allocateSymbols(self: *MachO) !void { | ... | @@ -2310,24 +1889,6 @@ fn allocateSymbols(self: *MachO) !void { |
| 2310 | } | 1889 | } |
| 2311 | } | 1890 | } |
| 2312 | | 1891 | |
| 2313 | fn shiftLocalsByOffset(self: *MachO, match: MatchingSection, offset: i64) !void { | | |
| 2314 | var atom = self.atoms.get(match) orelse return; | | |
| 2315 | | | |
| 2316 | while (true) { | | |
| 2317 | const atom_sym = atom.getSymbolPtr(self); | | |
| 2318 | atom_sym.n_value = @intCast(u64, @intCast(i64, atom_sym.n_value) + offset); | | |
| 2319 | | | |
| 2320 | for (atom.contained.items) |sym_at_off| { | | |
| 2321 | const contained_sym = self.getSymbolPtr(.{ .sym_index = sym_at_off.sym_index, .file = atom.file }); | | |
| 2322 | contained_sym.n_value = @intCast(u64, @intCast(i64, contained_sym.n_value) + offset); | | |
| 2323 | } | | |
| 2324 | | | |
| 2325 | if (atom.prev) |prev| { | | |
| 2326 | atom = prev; | | |
| 2327 | } else break; | | |
| 2328 | } | | |
| 2329 | } | | |
| 2330 | | | |
| 2331 | fn allocateSpecialSymbols(self: *MachO) !void { | 1892 | fn allocateSpecialSymbols(self: *MachO) !void { |
| 2332 | for (&[_][]const u8{ | 1893 | for (&[_][]const u8{ |
| 2333 | "___dso_handle", | 1894 | "___dso_handle", |
| ... | @@ -2336,16 +1897,13 @@ fn allocateSpecialSymbols(self: *MachO) !void { | ... | @@ -2336,16 +1897,13 @@ fn allocateSpecialSymbols(self: *MachO) !void { |
| 2336 | const global = self.globals.get(name) orelse continue; | 1897 | const global = self.globals.get(name) orelse continue; |
| 2337 | if (global.file != null) continue; | 1898 | if (global.file != null) continue; |
| 2338 | const sym = self.getSymbolPtr(global); | 1899 | const sym = self.getSymbolPtr(global); |
| 2339 | const seg = self.load_commands.items[self.text_segment_cmd_index.?].segment; | 1900 | const seg = self.segments.items[self.text_segment_cmd_index.?]; |
| 2340 | sym.n_sect = self.getSectionOrdinal(.{ | 1901 | sym.n_sect = 1; |
| 2341 | .seg = self.text_segment_cmd_index.?, | 1902 | sym.n_value = seg.vmaddr; |
| 2342 | .sect = 0, | | |
| 2343 | }); | | |
| 2344 | sym.n_value = seg.inner.vmaddr; | | |
| 2345 | | 1903 | |
| 2346 | log.debug("allocating {s} at the start of {s}", .{ | 1904 | log.debug("allocating {s} at the start of {s}", .{ |
| 2347 | name, | 1905 | name, |
| 2348 | seg.inner.segName(), | 1906 | seg.segName(), |
| 2349 | }); | 1907 | }); |
| 2350 | } | 1908 | } |
| 2351 | } | 1909 | } |
| ... | @@ -2353,18 +1911,21 @@ fn allocateSpecialSymbols(self: *MachO) !void { | ... | @@ -2353,18 +1911,21 @@ fn allocateSpecialSymbols(self: *MachO) !void { |
| 2353 | fn writeAtomsOneShot(self: *MachO) !void { | 1911 | fn writeAtomsOneShot(self: *MachO) !void { |
| 2354 | assert(self.mode == .one_shot); | 1912 | assert(self.mode == .one_shot); |
| 2355 | | 1913 | |
| 2356 | var it = self.atoms.iterator(); | 1914 | const gpa = self.base.allocator; |
| 2357 | while (it.next()) |entry| { | 1915 | const slice = self.sections.slice(); |
| 2358 | const sect = self.getSection(entry.key_ptr.*); | 1916 | |
| 2359 | var atom: *Atom = entry.value_ptr.*; | 1917 | for (slice.items(.last_atom)) |last_atom, sect_id| { |
| | 1918 | const header = slice.items(.header)[sect_id]; |
| | 1919 | if (header.size == 0) continue; |
| | 1920 | var atom = last_atom.?; |
| 2360 | | 1921 | |
| 2361 | if (sect.flags == macho.S_ZEROFILL or sect.flags == macho.S_THREAD_LOCAL_ZEROFILL) continue; | 1922 | if (header.isZerofill()) continue; |
| 2362 | | 1923 | |
| 2363 | var buffer = std.ArrayList(u8).init(self.base.allocator); | 1924 | var buffer = std.ArrayList(u8).init(gpa); |
| 2364 | defer buffer.deinit(); | 1925 | defer buffer.deinit(); |
| 2365 | try buffer.ensureTotalCapacity(math.cast(usize, sect.size) orelse return error.Overflow); | 1926 | try buffer.ensureTotalCapacity(math.cast(usize, header.size) orelse return error.Overflow); |
| 2366 | | 1927 | |
| 2367 | log.debug("writing atoms in {s},{s}", .{ sect.segName(), sect.sectName() }); | 1928 | log.debug("writing atoms in {s},{s}", .{ header.segName(), header.sectName() }); |
| 2368 | | 1929 | |
| 2369 | while (atom.prev) |prev| { | 1930 | while (atom.prev) |prev| { |
| 2370 | atom = prev; | 1931 | atom = prev; |
| ... | @@ -2399,18 +1960,18 @@ fn writeAtomsOneShot(self: *MachO) !void { | ... | @@ -2399,18 +1960,18 @@ fn writeAtomsOneShot(self: *MachO) !void { |
| 2399 | if (atom.next) |next| { | 1960 | if (atom.next) |next| { |
| 2400 | atom = next; | 1961 | atom = next; |
| 2401 | } else { | 1962 | } else { |
| 2402 | assert(buffer.items.len == sect.size); | 1963 | assert(buffer.items.len == header.size); |
| 2403 | log.debug(" (writing at file offset 0x{x})", .{sect.offset}); | 1964 | log.debug(" (writing at file offset 0x{x})", .{header.offset}); |
| 2404 | try self.base.file.?.pwriteAll(buffer.items, sect.offset); | 1965 | try self.base.file.?.pwriteAll(buffer.items, header.offset); |
| 2405 | break; | 1966 | break; |
| 2406 | } | 1967 | } |
| 2407 | } | 1968 | } |
| 2408 | } | 1969 | } |
| 2409 | } | 1970 | } |
| 2410 | | 1971 | |
| 2411 | fn writePadding(self: *MachO, match: MatchingSection, size: usize, writer: anytype) !void { | 1972 | fn writePadding(self: *MachO, sect_id: u8, size: usize, writer: anytype) !void { |
| 2412 | const is_code = match.seg == self.text_segment_cmd_index.? and match.sect == self.text_section_index.?; | 1973 | const header = self.sections.items(.header)[sect_id]; |
| 2413 | const min_alignment: u3 = if (!is_code) | 1974 | const min_alignment: u3 = if (!header.isCode()) |
| 2414 | 1 | 1975 | 1 |
| 2415 | else switch (self.base.options.target.cpu.arch) { | 1976 | else switch (self.base.options.target.cpu.arch) { |
| 2416 | .aarch64 => @sizeOf(u32), | 1977 | .aarch64 => @sizeOf(u32), |
| ... | @@ -2421,7 +1982,7 @@ fn writePadding(self: *MachO, match: MatchingSection, size: usize, writer: anyty | ... | @@ -2421,7 +1982,7 @@ fn writePadding(self: *MachO, match: MatchingSection, size: usize, writer: anyty |
| 2421 | const len = @divExact(size, min_alignment); | 1982 | const len = @divExact(size, min_alignment); |
| 2422 | var i: usize = 0; | 1983 | var i: usize = 0; |
| 2423 | while (i < len) : (i += 1) { | 1984 | while (i < len) : (i += 1) { |
| 2424 | if (!is_code) { | 1985 | if (!header.isCode()) { |
| 2425 | try writer.writeByte(0); | 1986 | try writer.writeByte(0); |
| 2426 | } else switch (self.base.options.target.cpu.arch) { | 1987 | } else switch (self.base.options.target.cpu.arch) { |
| 2427 | .aarch64 => { | 1988 | .aarch64 => { |
| ... | @@ -2439,20 +2000,19 @@ fn writePadding(self: *MachO, match: MatchingSection, size: usize, writer: anyty | ... | @@ -2439,20 +2000,19 @@ fn writePadding(self: *MachO, match: MatchingSection, size: usize, writer: anyty |
| 2439 | fn writeAtomsIncremental(self: *MachO) !void { | 2000 | fn writeAtomsIncremental(self: *MachO) !void { |
| 2440 | assert(self.mode == .incremental); | 2001 | assert(self.mode == .incremental); |
| 2441 | | 2002 | |
| 2442 | var it = self.atoms.iterator(); | 2003 | const slice = self.sections.slice(); |
| 2443 | while (it.next()) |entry| { | 2004 | for (slice.items(.last_atom)) |last, i| { |
| 2444 | const match = entry.key_ptr.*; | 2005 | var atom: *Atom = last orelse continue; |
| 2445 | const sect = self.getSection(match); | 2006 | const sect_i = @intCast(u8, i); |
| 2446 | var atom: *Atom = entry.value_ptr.*; | 2007 | const header = slice.items(.header)[sect_i]; |
| 2447 | | 2008 | |
| 2448 | // TODO handle zerofill in stage2 | 2009 | if (header.isZerofill()) continue; |
| 2449 | // if (sect.flags == macho.S_ZEROFILL or sect.flags == macho.S_THREAD_LOCAL_ZEROFILL) continue; | | |
| 2450 | | 2010 | |
| 2451 | log.debug("writing atoms in {s},{s}", .{ sect.segName(), sect.sectName() }); | 2011 | log.debug("writing atoms in {s},{s}", .{ header.segName(), header.sectName() }); |
| 2452 | | 2012 | |
| 2453 | while (true) { | 2013 | while (true) { |
| 2454 | if (atom.dirty) { | 2014 | if (atom.dirty) { |
| 2455 | try self.writeAtom(atom, match); | 2015 | try self.writeAtom(atom, sect_i); |
| 2456 | atom.dirty = false; | 2016 | atom.dirty = false; |
| 2457 | } | 2017 | } |
| 2458 | | 2018 | |
| ... | @@ -2503,10 +2063,7 @@ pub fn createGotAtom(self: *MachO, target: SymbolWithLoc) !*Atom { | ... | @@ -2503,10 +2063,7 @@ pub fn createGotAtom(self: *MachO, target: SymbolWithLoc) !*Atom { |
| 2503 | try self.managed_atoms.append(gpa, atom); | 2063 | try self.managed_atoms.append(gpa, atom); |
| 2504 | try self.atom_by_index_table.putNoClobber(gpa, sym_index, atom); | 2064 | try self.atom_by_index_table.putNoClobber(gpa, sym_index, atom); |
| 2505 | | 2065 | |
| 2506 | try self.allocateAtomCommon(atom, .{ | 2066 | try self.allocateAtomCommon(atom, self.got_section_index.?); |
| 2507 | .seg = self.data_const_segment_cmd_index.?, | | |
| 2508 | .sect = self.got_section_index.?, | | |
| 2509 | }); | | |
| 2510 | | 2067 | |
| 2511 | return atom; | 2068 | return atom; |
| 2512 | } | 2069 | } |
| ... | @@ -2535,7 +2092,7 @@ pub fn createTlvPtrAtom(self: *MachO, target: SymbolWithLoc) !*Atom { | ... | @@ -2535,7 +2092,7 @@ pub fn createTlvPtrAtom(self: *MachO, target: SymbolWithLoc) !*Atom { |
| 2535 | try self.managed_atoms.append(gpa, atom); | 2092 | try self.managed_atoms.append(gpa, atom); |
| 2536 | try self.atom_by_index_table.putNoClobber(gpa, sym_index, atom); | 2093 | try self.atom_by_index_table.putNoClobber(gpa, sym_index, atom); |
| 2537 | | 2094 | |
| 2538 | const match = (try self.getMatchingSection(.{ | 2095 | const match = (try self.getOutputSection(.{ |
| 2539 | .segname = makeStaticString("__DATA"), | 2096 | .segname = makeStaticString("__DATA"), |
| 2540 | .sectname = makeStaticString("__thread_ptrs"), | 2097 | .sectname = makeStaticString("__thread_ptrs"), |
| 2541 | .flags = macho.S_THREAD_LOCAL_VARIABLE_POINTERS, | 2098 | .flags = macho.S_THREAD_LOCAL_VARIABLE_POINTERS, |
| ... | @@ -2561,10 +2118,7 @@ fn createDyldPrivateAtom(self: *MachO) !void { | ... | @@ -2561,10 +2118,7 @@ fn createDyldPrivateAtom(self: *MachO) !void { |
| 2561 | const atom = try MachO.createEmptyAtom(gpa, sym_index, @sizeOf(u64), 3); | 2118 | const atom = try MachO.createEmptyAtom(gpa, sym_index, @sizeOf(u64), 3); |
| 2562 | self.dyld_private_atom = atom; | 2119 | self.dyld_private_atom = atom; |
| 2563 | | 2120 | |
| 2564 | try self.allocateAtomCommon(atom, .{ | 2121 | try self.allocateAtomCommon(atom, self.data_section_index.?); |
| 2565 | .seg = self.data_segment_cmd_index.?, | | |
| 2566 | .sect = self.data_section_index.?, | | |
| 2567 | }); | | |
| 2568 | | 2122 | |
| 2569 | try self.managed_atoms.append(gpa, atom); | 2123 | try self.managed_atoms.append(gpa, atom); |
| 2570 | try self.atom_by_index_table.putNoClobber(gpa, sym_index, atom); | 2124 | try self.atom_by_index_table.putNoClobber(gpa, sym_index, atom); |
| ... | @@ -2692,10 +2246,7 @@ fn createStubHelperPreambleAtom(self: *MachO) !void { | ... | @@ -2692,10 +2246,7 @@ fn createStubHelperPreambleAtom(self: *MachO) !void { |
| 2692 | } | 2246 | } |
| 2693 | self.stub_helper_preamble_atom = atom; | 2247 | self.stub_helper_preamble_atom = atom; |
| 2694 | | 2248 | |
| 2695 | try self.allocateAtomCommon(atom, .{ | 2249 | try self.allocateAtomCommon(atom, self.stub_helper_section_index.?); |
| 2696 | .seg = self.text_segment_cmd_index.?, | | |
| 2697 | .sect = self.stub_helper_section_index.?, | | |
| 2698 | }); | | |
| 2699 | | 2250 | |
| 2700 | try self.managed_atoms.append(gpa, atom); | 2251 | try self.managed_atoms.append(gpa, atom); |
| 2701 | try self.atom_by_index_table.putNoClobber(gpa, sym_index, atom); | 2252 | try self.atom_by_index_table.putNoClobber(gpa, sym_index, atom); |
| ... | @@ -2771,10 +2322,7 @@ pub fn createStubHelperAtom(self: *MachO) !*Atom { | ... | @@ -2771,10 +2322,7 @@ pub fn createStubHelperAtom(self: *MachO) !*Atom { |
| 2771 | try self.managed_atoms.append(gpa, atom); | 2322 | try self.managed_atoms.append(gpa, atom); |
| 2772 | try self.atom_by_index_table.putNoClobber(gpa, sym_index, atom); | 2323 | try self.atom_by_index_table.putNoClobber(gpa, sym_index, atom); |
| 2773 | | 2324 | |
| 2774 | try self.allocateAtomCommon(atom, .{ | 2325 | try self.allocateAtomCommon(atom, self.stub_helper_section_index.?); |
| 2775 | .seg = self.text_segment_cmd_index.?, | | |
| 2776 | .sect = self.stub_helper_section_index.?, | | |
| 2777 | }); | | |
| 2778 | | 2326 | |
| 2779 | return atom; | 2327 | return atom; |
| 2780 | } | 2328 | } |
| ... | @@ -2814,10 +2362,7 @@ pub fn createLazyPointerAtom(self: *MachO, stub_sym_index: u32, target: SymbolWi | ... | @@ -2814,10 +2362,7 @@ pub fn createLazyPointerAtom(self: *MachO, stub_sym_index: u32, target: SymbolWi |
| 2814 | try self.managed_atoms.append(gpa, atom); | 2362 | try self.managed_atoms.append(gpa, atom); |
| 2815 | try self.atom_by_index_table.putNoClobber(gpa, sym_index, atom); | 2363 | try self.atom_by_index_table.putNoClobber(gpa, sym_index, atom); |
| 2816 | | 2364 | |
| 2817 | try self.allocateAtomCommon(atom, .{ | 2365 | try self.allocateAtomCommon(atom, self.la_symbol_ptr_section_index.?); |
| 2818 | .seg = self.data_segment_cmd_index.?, | | |
| 2819 | .sect = self.la_symbol_ptr_section_index.?, | | |
| 2820 | }); | | |
| 2821 | | 2366 | |
| 2822 | return atom; | 2367 | return atom; |
| 2823 | } | 2368 | } |
| ... | @@ -2896,10 +2441,7 @@ pub fn createStubAtom(self: *MachO, laptr_sym_index: u32) !*Atom { | ... | @@ -2896,10 +2441,7 @@ pub fn createStubAtom(self: *MachO, laptr_sym_index: u32) !*Atom { |
| 2896 | try self.managed_atoms.append(gpa, atom); | 2441 | try self.managed_atoms.append(gpa, atom); |
| 2897 | try self.atom_by_index_table.putNoClobber(gpa, sym_index, atom); | 2442 | try self.atom_by_index_table.putNoClobber(gpa, sym_index, atom); |
| 2898 | | 2443 | |
| 2899 | try self.allocateAtomCommon(atom, .{ | 2444 | try self.allocateAtomCommon(atom, self.stubs_section_index.?); |
| 2900 | .seg = self.text_segment_cmd_index.?, | | |
| 2901 | .sect = self.stubs_section_index.?, | | |
| 2902 | }); | | |
| 2903 | | 2445 | |
| 2904 | return atom; | 2446 | return atom; |
| 2905 | } | 2447 | } |
| ... | @@ -2917,19 +2459,18 @@ fn createTentativeDefAtoms(self: *MachO) !void { | ... | @@ -2917,19 +2459,18 @@ fn createTentativeDefAtoms(self: *MachO) !void { |
| 2917 | | 2459 | |
| 2918 | // Convert any tentative definition into a regular symbol and allocate | 2460 | // Convert any tentative definition into a regular symbol and allocate |
| 2919 | // text blocks for each tentative definition. | 2461 | // text blocks for each tentative definition. |
| 2920 | const match = MatchingSection{ | | |
| 2921 | .seg = self.data_segment_cmd_index.?, | | |
| 2922 | .sect = self.bss_section_index.?, | | |
| 2923 | }; | | |
| 2924 | _ = try self.section_ordinals.getOrPut(gpa, match); | | |
| 2925 | | | |
| 2926 | const size = sym.n_value; | 2462 | const size = sym.n_value; |
| 2927 | const alignment = (sym.n_desc >> 8) & 0x0f; | 2463 | const alignment = (sym.n_desc >> 8) & 0x0f; |
| | 2464 | const n_sect = (try self.getOutputSection(.{ |
| | 2465 | .segname = makeStaticString("__DATA"), |
| | 2466 | .sectname = makeStaticString("__bss"), |
| | 2467 | .flags = macho.S_ZEROFILL, |
| | 2468 | })).?; |
| 2928 | | 2469 | |
| 2929 | sym.* = .{ | 2470 | sym.* = .{ |
| 2930 | .n_strx = sym.n_strx, | 2471 | .n_strx = sym.n_strx, |
| 2931 | .n_type = macho.N_SECT | macho.N_EXT, | 2472 | .n_type = macho.N_SECT | macho.N_EXT, |
| 2932 | .n_sect = 0, | 2473 | .n_sect = n_sect, |
| 2933 | .n_desc = 0, | 2474 | .n_desc = 0, |
| 2934 | .n_value = 0, | 2475 | .n_value = 0, |
| 2935 | }; | 2476 | }; |
| ... | @@ -2937,7 +2478,7 @@ fn createTentativeDefAtoms(self: *MachO) !void { | ... | @@ -2937,7 +2478,7 @@ fn createTentativeDefAtoms(self: *MachO) !void { |
| 2937 | const atom = try MachO.createEmptyAtom(gpa, global.sym_index, size, alignment); | 2478 | const atom = try MachO.createEmptyAtom(gpa, global.sym_index, size, alignment); |
| 2938 | atom.file = global.file; | 2479 | atom.file = global.file; |
| 2939 | | 2480 | |
| 2940 | try self.allocateAtomCommon(atom, match); | 2481 | try self.allocateAtomCommon(atom, n_sect); |
| 2941 | | 2482 | |
| 2942 | if (global.file) |file| { | 2483 | if (global.file) |file| { |
| 2943 | const object = &self.objects.items[file]; | 2484 | const object = &self.objects.items[file]; |
| ... | @@ -3060,7 +2601,8 @@ fn resolveGlobalSymbol(self: *MachO, current: SymbolWithLoc) !void { | ... | @@ -3060,7 +2601,8 @@ fn resolveGlobalSymbol(self: *MachO, current: SymbolWithLoc) !void { |
| 3060 | gop.value_ptr.* = current; | 2601 | gop.value_ptr.* = current; |
| 3061 | } | 2602 | } |
| 3062 | | 2603 | |
| 3063 | fn resolveSymbolsInObject(self: *MachO, object: *Object, object_id: u16) !void { | 2604 | fn resolveSymbolsInObject(self: *MachO, object_id: u16) !void { |
| | 2605 | const object = &self.objects.items[object_id]; |
| 3064 | log.debug("resolving symbols in '{s}'", .{object.name}); | 2606 | log.debug("resolving symbols in '{s}'", .{object.name}); |
| 3065 | | 2607 | |
| 3066 | for (object.symtab.items) |sym, index| { | 2608 | for (object.symtab.items) |sym, index| { |
| ... | @@ -3115,6 +2657,8 @@ fn resolveSymbolsInObject(self: *MachO, object: *Object, object_id: u16) !void { | ... | @@ -3115,6 +2657,8 @@ fn resolveSymbolsInObject(self: *MachO, object: *Object, object_id: u16) !void { |
| 3115 | fn resolveSymbolsInArchives(self: *MachO) !void { | 2657 | fn resolveSymbolsInArchives(self: *MachO) !void { |
| 3116 | if (self.archives.items.len == 0) return; | 2658 | if (self.archives.items.len == 0) return; |
| 3117 | | 2659 | |
| | 2660 | const gpa = self.base.allocator; |
| | 2661 | const cpu_arch = self.base.options.target.cpu.arch; |
| 3118 | var next_sym: usize = 0; | 2662 | var next_sym: usize = 0; |
| 3119 | loop: while (next_sym < self.unresolved.count()) { | 2663 | loop: while (next_sym < self.unresolved.count()) { |
| 3120 | const global = self.globals.values()[self.unresolved.keys()[next_sym]]; | 2664 | const global = self.globals.values()[self.unresolved.keys()[next_sym]]; |
| ... | @@ -3129,13 +2673,9 @@ fn resolveSymbolsInArchives(self: *MachO) !void { | ... | @@ -3129,13 +2673,9 @@ fn resolveSymbolsInArchives(self: *MachO) !void { |
| 3129 | assert(offsets.items.len > 0); | 2673 | assert(offsets.items.len > 0); |
| 3130 | | 2674 | |
| 3131 | const object_id = @intCast(u16, self.objects.items.len); | 2675 | const object_id = @intCast(u16, self.objects.items.len); |
| 3132 | const object = try self.objects.addOne(self.base.allocator); | 2676 | const object = try archive.parseObject(gpa, cpu_arch, offsets.items[0]); |
| 3133 | object.* = try archive.parseObject( | 2677 | try self.objects.append(gpa, object); |
| 3134 | self.base.allocator, | 2678 | try self.resolveSymbolsInObject(object_id); |
| 3135 | self.base.options.target.cpu.arch, | | |
| 3136 | offsets.items[0], | | |
| 3137 | ); | | |
| 3138 | try self.resolveSymbolsInObject(object, object_id); | | |
| 3139 | | 2679 | |
| 3140 | continue :loop; | 2680 | continue :loop; |
| 3141 | } | 2681 | } |
| ... | @@ -3159,7 +2699,6 @@ fn resolveSymbolsInDylibs(self: *MachO) !void { | ... | @@ -3159,7 +2699,6 @@ fn resolveSymbolsInDylibs(self: *MachO) !void { |
| 3159 | | 2699 | |
| 3160 | const dylib_id = @intCast(u16, id); | 2700 | const dylib_id = @intCast(u16, id); |
| 3161 | if (!self.referenced_dylibs.contains(dylib_id)) { | 2701 | if (!self.referenced_dylibs.contains(dylib_id)) { |
| 3162 | try self.addLoadDylibLC(dylib_id); | | |
| 3163 | try self.referenced_dylibs.putNoClobber(self.base.allocator, dylib_id, {}); | 2702 | try self.referenced_dylibs.putNoClobber(self.base.allocator, dylib_id, {}); |
| 3164 | } | 2703 | } |
| 3165 | | 2704 | |
| ... | @@ -3257,7 +2796,6 @@ fn resolveDyldStubBinder(self: *MachO) !void { | ... | @@ -3257,7 +2796,6 @@ fn resolveDyldStubBinder(self: *MachO) !void { |
| 3257 | | 2796 | |
| 3258 | const dylib_id = @intCast(u16, id); | 2797 | const dylib_id = @intCast(u16, id); |
| 3259 | if (!self.referenced_dylibs.contains(dylib_id)) { | 2798 | if (!self.referenced_dylibs.contains(dylib_id)) { |
| 3260 | try self.addLoadDylibLC(dylib_id); | | |
| 3261 | try self.referenced_dylibs.putNoClobber(self.base.allocator, dylib_id, {}); | 2799 | try self.referenced_dylibs.putNoClobber(self.base.allocator, dylib_id, {}); |
| 3262 | } | 2800 | } |
| 3263 | | 2801 | |
| ... | @@ -3280,47 +2818,192 @@ fn resolveDyldStubBinder(self: *MachO) !void { | ... | @@ -3280,47 +2818,192 @@ fn resolveDyldStubBinder(self: *MachO) !void { |
| 3280 | self.got_entries.items[got_index].sym_index = got_atom.sym_index; | 2818 | self.got_entries.items[got_index].sym_index = got_atom.sym_index; |
| 3281 | } | 2819 | } |
| 3282 | | 2820 | |
| 3283 | fn addLoadDylibLC(self: *MachO, id: u16) !void { | 2821 | fn writeDylinkerLC(ncmds: *u32, lc_writer: anytype) !void { |
| 3284 | const dylib = self.dylibs.items[id]; | 2822 | const name_len = mem.sliceTo(default_dyld_path, 0).len; |
| 3285 | const dylib_id = dylib.id orelse unreachable; | 2823 | const cmdsize = @intCast(u32, mem.alignForwardGeneric( |
| 3286 | var dylib_cmd = try macho.createLoadDylibCommand( | 2824 | u64, |
| 3287 | self.base.allocator, | 2825 | @sizeOf(macho.dylinker_command) + name_len, |
| 3288 | if (dylib.weak) .LOAD_WEAK_DYLIB else .LOAD_DYLIB, | 2826 | @sizeOf(u64), |
| 3289 | dylib_id.name, | 2827 | )); |
| 3290 | dylib_id.timestamp, | 2828 | try lc_writer.writeStruct(macho.dylinker_command{ |
| 3291 | dylib_id.current_version, | 2829 | .cmd = .LOAD_DYLINKER, |
| 3292 | dylib_id.compatibility_version, | 2830 | .cmdsize = cmdsize, |
| 3293 | ); | 2831 | .name = @sizeOf(macho.dylinker_command), |
| 3294 | errdefer dylib_cmd.deinit(self.base.allocator); | 2832 | }); |
| 3295 | try self.load_commands.append(self.base.allocator, .{ .dylib = dylib_cmd }); | 2833 | try lc_writer.writeAll(mem.sliceTo(default_dyld_path, 0)); |
| 3296 | self.load_commands_dirty = true; | 2834 | const padding = cmdsize - @sizeOf(macho.dylinker_command) - name_len; |
| | 2835 | if (padding > 0) { |
| | 2836 | try lc_writer.writeByteNTimes(0, padding); |
| | 2837 | } |
| | 2838 | ncmds.* += 1; |
| 3297 | } | 2839 | } |
| 3298 | | 2840 | |
| 3299 | fn addCodeSignatureLC(self: *MachO) !void { | 2841 | fn writeMainLC(self: *MachO, ncmds: *u32, lc_writer: anytype) !void { |
| 3300 | if (self.code_signature_cmd_index != null or self.code_signature == null) return; | 2842 | if (self.base.options.output_mode != .Exe) return; |
| 3301 | self.code_signature_cmd_index = @intCast(u16, self.load_commands.items.len); | 2843 | const seg = self.segments.items[self.text_segment_cmd_index.?]; |
| 3302 | try self.load_commands.append(self.base.allocator, .{ | 2844 | const global = try self.getEntryPoint(); |
| 3303 | .linkedit_data = .{ | 2845 | const sym = self.getSymbol(global); |
| 3304 | .cmd = .CODE_SIGNATURE, | 2846 | try lc_writer.writeStruct(macho.entry_point_command{ |
| 3305 | .cmdsize = @sizeOf(macho.linkedit_data_command), | 2847 | .cmd = .MAIN, |
| 3306 | .dataoff = 0, | 2848 | .cmdsize = @sizeOf(macho.entry_point_command), |
| 3307 | .datasize = 0, | 2849 | .entryoff = @intCast(u32, sym.n_value - seg.vmaddr), |
| | 2850 | .stacksize = self.base.options.stack_size_override orelse 0, |
| | 2851 | }); |
| | 2852 | ncmds.* += 1; |
| | 2853 | } |
| | 2854 | |
| | 2855 | const WriteDylibLCCtx = struct { |
| | 2856 | cmd: macho.LC, |
| | 2857 | name: []const u8, |
| | 2858 | timestamp: u32 = 2, |
| | 2859 | current_version: u32 = 0x10000, |
| | 2860 | compatibility_version: u32 = 0x10000, |
| | 2861 | }; |
| | 2862 | |
| | 2863 | fn writeDylibLC(ctx: WriteDylibLCCtx, ncmds: *u32, lc_writer: anytype) !void { |
| | 2864 | const name_len = ctx.name.len + 1; |
| | 2865 | const cmdsize = @intCast(u32, mem.alignForwardGeneric( |
| | 2866 | u64, |
| | 2867 | @sizeOf(macho.dylib_command) + name_len, |
| | 2868 | @sizeOf(u64), |
| | 2869 | )); |
| | 2870 | try lc_writer.writeStruct(macho.dylib_command{ |
| | 2871 | .cmd = ctx.cmd, |
| | 2872 | .cmdsize = cmdsize, |
| | 2873 | .dylib = .{ |
| | 2874 | .name = @sizeOf(macho.dylib_command), |
| | 2875 | .timestamp = ctx.timestamp, |
| | 2876 | .current_version = ctx.current_version, |
| | 2877 | .compatibility_version = ctx.compatibility_version, |
| 3308 | }, | 2878 | }, |
| 3309 | }); | 2879 | }); |
| 3310 | self.load_commands_dirty = true; | 2880 | try lc_writer.writeAll(ctx.name); |
| | 2881 | try lc_writer.writeByte(0); |
| | 2882 | const padding = cmdsize - @sizeOf(macho.dylib_command) - name_len; |
| | 2883 | if (padding > 0) { |
| | 2884 | try lc_writer.writeByteNTimes(0, padding); |
| | 2885 | } |
| | 2886 | ncmds.* += 1; |
| 3311 | } | 2887 | } |
| 3312 | | 2888 | |
| 3313 | fn setEntryPoint(self: *MachO) !void { | 2889 | fn writeDylibIdLC(self: *MachO, ncmds: *u32, lc_writer: anytype) !void { |
| 3314 | if (self.base.options.output_mode != .Exe) return; | 2890 | if (self.base.options.output_mode != .Lib) return; |
| | 2891 | const install_name = self.base.options.install_name orelse self.base.options.emit.?.sub_path; |
| | 2892 | const curr = self.base.options.version orelse std.builtin.Version{ |
| | 2893 | .major = 1, |
| | 2894 | .minor = 0, |
| | 2895 | .patch = 0, |
| | 2896 | }; |
| | 2897 | const compat = self.base.options.compatibility_version orelse std.builtin.Version{ |
| | 2898 | .major = 1, |
| | 2899 | .minor = 0, |
| | 2900 | .patch = 0, |
| | 2901 | }; |
| | 2902 | try writeDylibLC(.{ |
| | 2903 | .cmd = .ID_DYLIB, |
| | 2904 | .name = install_name, |
| | 2905 | .current_version = curr.major << 16 | curr.minor << 8 | curr.patch, |
| | 2906 | .compatibility_version = compat.major << 16 | compat.minor << 8 | compat.patch, |
| | 2907 | }, ncmds, lc_writer); |
| | 2908 | } |
| 3315 | | 2909 | |
| 3316 | const seg = self.load_commands.items[self.text_segment_cmd_index.?].segment; | 2910 | const RpathIterator = struct { |
| 3317 | const global = try self.getEntryPoint(); | 2911 | buffer: []const []const u8, |
| 3318 | const sym = self.getSymbol(global); | 2912 | table: std.StringHashMap(void), |
| 3319 | const ec = &self.load_commands.items[self.main_cmd_index.?].main; | 2913 | count: usize = 0, |
| 3320 | ec.entryoff = @intCast(u32, sym.n_value - seg.inner.vmaddr); | 2914 | |
| 3321 | ec.stacksize = self.base.options.stack_size_override orelse 0; | 2915 | fn init(gpa: Allocator, rpaths: []const []const u8) RpathIterator { |
| 3322 | self.entry_addr = sym.n_value; | 2916 | return .{ .buffer = rpaths, .table = std.StringHashMap(void).init(gpa) }; |
| 3323 | self.load_commands_dirty = true; | 2917 | } |
| | 2918 | |
| | 2919 | fn deinit(it: *RpathIterator) void { |
| | 2920 | it.table.deinit(); |
| | 2921 | } |
| | 2922 | |
| | 2923 | fn next(it: *RpathIterator) !?[]const u8 { |
| | 2924 | while (true) { |
| | 2925 | if (it.count >= it.buffer.len) return null; |
| | 2926 | const rpath = it.buffer[it.count]; |
| | 2927 | it.count += 1; |
| | 2928 | const gop = try it.table.getOrPut(rpath); |
| | 2929 | if (gop.found_existing) continue; |
| | 2930 | return rpath; |
| | 2931 | } |
| | 2932 | } |
| | 2933 | }; |
| | 2934 | |
| | 2935 | fn writeRpathLCs(self: *MachO, ncmds: *u32, lc_writer: anytype) !void { |
| | 2936 | const gpa = self.base.allocator; |
| | 2937 | |
| | 2938 | var it = RpathIterator.init(gpa, self.base.options.rpath_list); |
| | 2939 | defer it.deinit(); |
| | 2940 | |
| | 2941 | while (try it.next()) |rpath| { |
| | 2942 | const rpath_len = rpath.len + 1; |
| | 2943 | const cmdsize = @intCast(u32, mem.alignForwardGeneric( |
| | 2944 | u64, |
| | 2945 | @sizeOf(macho.rpath_command) + rpath_len, |
| | 2946 | @sizeOf(u64), |
| | 2947 | )); |
| | 2948 | try lc_writer.writeStruct(macho.rpath_command{ |
| | 2949 | .cmdsize = cmdsize, |
| | 2950 | .path = @sizeOf(macho.rpath_command), |
| | 2951 | }); |
| | 2952 | try lc_writer.writeAll(rpath); |
| | 2953 | try lc_writer.writeByte(0); |
| | 2954 | const padding = cmdsize - @sizeOf(macho.rpath_command) - rpath_len; |
| | 2955 | if (padding > 0) { |
| | 2956 | try lc_writer.writeByteNTimes(0, padding); |
| | 2957 | } |
| | 2958 | ncmds.* += 1; |
| | 2959 | } |
| | 2960 | } |
| | 2961 | |
| | 2962 | fn writeBuildVersionLC(self: *MachO, ncmds: *u32, lc_writer: anytype) !void { |
| | 2963 | const cmdsize = @sizeOf(macho.build_version_command) + @sizeOf(macho.build_tool_version); |
| | 2964 | const platform_version = blk: { |
| | 2965 | const ver = self.base.options.target.os.version_range.semver.min; |
| | 2966 | const platform_version = ver.major << 16 | ver.minor << 8; |
| | 2967 | break :blk platform_version; |
| | 2968 | }; |
| | 2969 | const sdk_version = if (self.base.options.native_darwin_sdk) |sdk| blk: { |
| | 2970 | const ver = sdk.version; |
| | 2971 | const sdk_version = ver.major << 16 | ver.minor << 8; |
| | 2972 | break :blk sdk_version; |
| | 2973 | } else platform_version; |
| | 2974 | const is_simulator_abi = self.base.options.target.abi == .simulator; |
| | 2975 | try lc_writer.writeStruct(macho.build_version_command{ |
| | 2976 | .cmdsize = cmdsize, |
| | 2977 | .platform = switch (self.base.options.target.os.tag) { |
| | 2978 | .macos => .MACOS, |
| | 2979 | .ios => if (is_simulator_abi) macho.PLATFORM.IOSSIMULATOR else macho.PLATFORM.IOS, |
| | 2980 | .watchos => if (is_simulator_abi) macho.PLATFORM.WATCHOSSIMULATOR else macho.PLATFORM.WATCHOS, |
| | 2981 | .tvos => if (is_simulator_abi) macho.PLATFORM.TVOSSIMULATOR else macho.PLATFORM.TVOS, |
| | 2982 | else => unreachable, |
| | 2983 | }, |
| | 2984 | .minos = platform_version, |
| | 2985 | .sdk = sdk_version, |
| | 2986 | .ntools = 1, |
| | 2987 | }); |
| | 2988 | try lc_writer.writeAll(mem.asBytes(&macho.build_tool_version{ |
| | 2989 | .tool = .LD, |
| | 2990 | .version = 0x0, |
| | 2991 | })); |
| | 2992 | ncmds.* += 1; |
| | 2993 | } |
| | 2994 | |
| | 2995 | fn writeLoadDylibLCs(self: *MachO, ncmds: *u32, lc_writer: anytype) !void { |
| | 2996 | for (self.referenced_dylibs.keys()) |id| { |
| | 2997 | const dylib = self.dylibs.items[id]; |
| | 2998 | const dylib_id = dylib.id orelse unreachable; |
| | 2999 | try writeDylibLC(.{ |
| | 3000 | .cmd = if (dylib.weak) .LOAD_WEAK_DYLIB else .LOAD_DYLIB, |
| | 3001 | .name = dylib_id.name, |
| | 3002 | .timestamp = dylib_id.timestamp, |
| | 3003 | .current_version = dylib_id.current_version, |
| | 3004 | .compatibility_version = dylib_id.compatibility_version, |
| | 3005 | }, ncmds, lc_writer); |
| | 3006 | } |
| 3324 | } | 3007 | } |
| 3325 | | 3008 | |
| 3326 | pub fn deinit(self: *MachO) void { | 3009 | pub fn deinit(self: *MachO) void { |
| ... | @@ -3334,7 +3017,6 @@ pub fn deinit(self: *MachO) void { | ... | @@ -3334,7 +3017,6 @@ pub fn deinit(self: *MachO) void { |
| 3334 | d_sym.deinit(gpa); | 3017 | d_sym.deinit(gpa); |
| 3335 | } | 3018 | } |
| 3336 | | 3019 | |
| 3337 | self.section_ordinals.deinit(gpa); | | |
| 3338 | self.tlv_ptr_entries.deinit(gpa); | 3020 | self.tlv_ptr_entries.deinit(gpa); |
| 3339 | self.tlv_ptr_entries_free_list.deinit(gpa); | 3021 | self.tlv_ptr_entries_free_list.deinit(gpa); |
| 3340 | self.tlv_ptr_entries_table.deinit(gpa); | 3022 | self.tlv_ptr_entries_table.deinit(gpa); |
| ... | @@ -3371,24 +3053,19 @@ pub fn deinit(self: *MachO) void { | ... | @@ -3371,24 +3053,19 @@ pub fn deinit(self: *MachO) void { |
| 3371 | self.dylibs_map.deinit(gpa); | 3053 | self.dylibs_map.deinit(gpa); |
| 3372 | self.referenced_dylibs.deinit(gpa); | 3054 | self.referenced_dylibs.deinit(gpa); |
| 3373 | | 3055 | |
| 3374 | for (self.load_commands.items) |*lc| { | 3056 | self.segments.deinit(gpa); |
| 3375 | lc.deinit(gpa); | 3057 | |
| | 3058 | for (self.sections.items(.free_list)) |*list| { |
| | 3059 | list.deinit(gpa); |
| 3376 | } | 3060 | } |
| 3377 | self.load_commands.deinit(gpa); | 3061 | self.sections.deinit(gpa); |
| 3378 | | 3062 | |
| 3379 | for (self.managed_atoms.items) |atom| { | 3063 | for (self.managed_atoms.items) |atom| { |
| 3380 | atom.deinit(gpa); | 3064 | atom.deinit(gpa); |
| 3381 | gpa.destroy(atom); | 3065 | gpa.destroy(atom); |
| 3382 | } | 3066 | } |
| 3383 | self.managed_atoms.deinit(gpa); | 3067 | self.managed_atoms.deinit(gpa); |
| 3384 | self.atoms.deinit(gpa); | 3068 | |
| 3385 | { | | |
| 3386 | var it = self.atom_free_lists.valueIterator(); | | |
| 3387 | while (it.next()) |free_list| { | | |
| 3388 | free_list.deinit(gpa); | | |
| 3389 | } | | |
| 3390 | self.atom_free_lists.deinit(gpa); | | |
| 3391 | } | | |
| 3392 | if (self.base.options.module) |mod| { | 3069 | if (self.base.options.module) |mod| { |
| 3393 | for (self.decls.keys()) |decl_index| { | 3070 | for (self.decls.keys()) |decl_index| { |
| 3394 | const decl = mod.declPtr(decl_index); | 3071 | const decl = mod.declPtr(decl_index); |
| ... | @@ -3408,34 +3085,24 @@ pub fn deinit(self: *MachO) void { | ... | @@ -3408,34 +3085,24 @@ pub fn deinit(self: *MachO) void { |
| 3408 | } | 3085 | } |
| 3409 | | 3086 | |
| 3410 | self.atom_by_index_table.deinit(gpa); | 3087 | self.atom_by_index_table.deinit(gpa); |
| 3411 | | | |
| 3412 | if (self.code_signature) |*csig| { | | |
| 3413 | csig.deinit(gpa); | | |
| 3414 | } | | |
| 3415 | } | 3088 | } |
| 3416 | | 3089 | |
| 3417 | pub fn closeFiles(self: MachO) void { | 3090 | pub fn closeFiles(self: MachO) void { |
| 3418 | for (self.objects.items) |object| { | | |
| 3419 | object.file.close(); | | |
| 3420 | } | | |
| 3421 | for (self.archives.items) |archive| { | 3091 | for (self.archives.items) |archive| { |
| 3422 | archive.file.close(); | 3092 | archive.file.close(); |
| 3423 | } | 3093 | } |
| 3424 | for (self.dylibs.items) |dylib| { | | |
| 3425 | dylib.file.close(); | | |
| 3426 | } | | |
| 3427 | if (self.d_sym) |ds| { | 3094 | if (self.d_sym) |ds| { |
| 3428 | ds.file.close(); | 3095 | ds.file.close(); |
| 3429 | } | 3096 | } |
| 3430 | } | 3097 | } |
| 3431 | | 3098 | |
| 3432 | fn freeAtom(self: *MachO, atom: *Atom, match: MatchingSection, owns_atom: bool) void { | 3099 | fn freeAtom(self: *MachO, atom: *Atom, sect_id: u8, owns_atom: bool) void { |
| 3433 | log.debug("freeAtom {*}", .{atom}); | 3100 | log.debug("freeAtom {*}", .{atom}); |
| 3434 | if (!owns_atom) { | 3101 | if (!owns_atom) { |
| 3435 | atom.deinit(self.base.allocator); | 3102 | atom.deinit(self.base.allocator); |
| 3436 | } | 3103 | } |
| 3437 | | 3104 | |
| 3438 | const free_list = self.atom_free_lists.getPtr(match).?; | 3105 | const free_list = &self.sections.items(.free_list)[sect_id]; |
| 3439 | var already_have_free_list_node = false; | 3106 | var already_have_free_list_node = false; |
| 3440 | { | 3107 | { |
| 3441 | var i: usize = 0; | 3108 | var i: usize = 0; |
| ... | @@ -3452,13 +3119,14 @@ fn freeAtom(self: *MachO, atom: *Atom, match: MatchingSection, owns_atom: bool) | ... | @@ -3452,13 +3119,14 @@ fn freeAtom(self: *MachO, atom: *Atom, match: MatchingSection, owns_atom: bool) |
| 3452 | } | 3119 | } |
| 3453 | } | 3120 | } |
| 3454 | | 3121 | |
| 3455 | if (self.atoms.getPtr(match)) |last_atom| { | 3122 | const maybe_last_atom = &self.sections.items(.last_atom)[sect_id]; |
| 3456 | if (last_atom.* == atom) { | 3123 | if (maybe_last_atom.*) |last_atom| { |
| | 3124 | if (last_atom == atom) { |
| 3457 | if (atom.prev) |prev| { | 3125 | if (atom.prev) |prev| { |
| 3458 | // TODO shrink the section size here | 3126 | // TODO shrink the section size here |
| 3459 | last_atom.* = prev; | 3127 | maybe_last_atom.* = prev; |
| 3460 | } else { | 3128 | } else { |
| 3461 | _ = self.atoms.fetchRemove(match); | 3129 | maybe_last_atom.* = null; |
| 3462 | } | 3130 | } |
| 3463 | } | 3131 | } |
| 3464 | } | 3132 | } |
| ... | @@ -3486,21 +3154,21 @@ fn freeAtom(self: *MachO, atom: *Atom, match: MatchingSection, owns_atom: bool) | ... | @@ -3486,21 +3154,21 @@ fn freeAtom(self: *MachO, atom: *Atom, match: MatchingSection, owns_atom: bool) |
| 3486 | } | 3154 | } |
| 3487 | } | 3155 | } |
| 3488 | | 3156 | |
| 3489 | fn shrinkAtom(self: *MachO, atom: *Atom, new_block_size: u64, match: MatchingSection) void { | 3157 | fn shrinkAtom(self: *MachO, atom: *Atom, new_block_size: u64, sect_id: u8) void { |
| 3490 | _ = self; | 3158 | _ = self; |
| 3491 | _ = atom; | 3159 | _ = atom; |
| 3492 | _ = new_block_size; | 3160 | _ = new_block_size; |
| 3493 | _ = match; | 3161 | _ = sect_id; |
| 3494 | // TODO check the new capacity, and if it crosses the size threshold into a big enough | 3162 | // TODO check the new capacity, and if it crosses the size threshold into a big enough |
| 3495 | // capacity, insert a free list node for it. | 3163 | // capacity, insert a free list node for it. |
| 3496 | } | 3164 | } |
| 3497 | | 3165 | |
| 3498 | fn growAtom(self: *MachO, atom: *Atom, new_atom_size: u64, alignment: u64, match: MatchingSection) !u64 { | 3166 | fn growAtom(self: *MachO, atom: *Atom, new_atom_size: u64, alignment: u64, sect_id: u8) !u64 { |
| 3499 | const sym = atom.getSymbol(self); | 3167 | const sym = atom.getSymbol(self); |
| 3500 | const align_ok = mem.alignBackwardGeneric(u64, sym.n_value, alignment) == sym.n_value; | 3168 | const align_ok = mem.alignBackwardGeneric(u64, sym.n_value, alignment) == sym.n_value; |
| 3501 | const need_realloc = !align_ok or new_atom_size > atom.capacity(self); | 3169 | const need_realloc = !align_ok or new_atom_size > atom.capacity(self); |
| 3502 | if (!need_realloc) return sym.n_value; | 3170 | if (!need_realloc) return sym.n_value; |
| 3503 | return self.allocateAtom(atom, new_atom_size, alignment, match); | 3171 | return self.allocateAtom(atom, new_atom_size, alignment, sect_id); |
| 3504 | } | 3172 | } |
| 3505 | | 3173 | |
| 3506 | fn allocateSymbol(self: *MachO) !u32 { | 3174 | fn allocateSymbol(self: *MachO) !u32 { |
| ... | @@ -3671,10 +3339,11 @@ pub fn updateFunc(self: *MachO, module: *Module, func: *Module.Fn, air: Air, liv | ... | @@ -3671,10 +3339,11 @@ pub fn updateFunc(self: *MachO, module: *Module, func: *Module.Fn, air: Air, liv |
| 3671 | } | 3339 | } |
| 3672 | | 3340 | |
| 3673 | pub fn lowerUnnamedConst(self: *MachO, typed_value: TypedValue, decl_index: Module.Decl.Index) !u32 { | 3341 | pub fn lowerUnnamedConst(self: *MachO, typed_value: TypedValue, decl_index: Module.Decl.Index) !u32 { |
| 3674 | var code_buffer = std.ArrayList(u8).init(self.base.allocator); | 3342 | const gpa = self.base.allocator; |
| | 3343 | |
| | 3344 | var code_buffer = std.ArrayList(u8).init(gpa); |
| 3675 | defer code_buffer.deinit(); | 3345 | defer code_buffer.deinit(); |
| 3676 | | 3346 | |
| 3677 | const gpa = self.base.allocator; | | |
| 3678 | const module = self.base.options.module.?; | 3347 | const module = self.base.options.module.?; |
| 3679 | const gop = try self.unnamed_const_atoms.getOrPut(gpa, decl_index); | 3348 | const gop = try self.unnamed_const_atoms.getOrPut(gpa, decl_index); |
| 3680 | if (!gop.found_existing) { | 3349 | if (!gop.found_existing) { |
| ... | @@ -3725,25 +3394,25 @@ pub fn lowerUnnamedConst(self: *MachO, typed_value: TypedValue, decl_index: Modu | ... | @@ -3725,25 +3394,25 @@ pub fn lowerUnnamedConst(self: *MachO, typed_value: TypedValue, decl_index: Modu |
| 3725 | atom.code.clearRetainingCapacity(); | 3394 | atom.code.clearRetainingCapacity(); |
| 3726 | try atom.code.appendSlice(gpa, code); | 3395 | try atom.code.appendSlice(gpa, code); |
| 3727 | | 3396 | |
| 3728 | const match = try self.getMatchingSectionAtom( | 3397 | const sect_id = try self.getOutputSectionAtom( |
| 3729 | atom, | 3398 | atom, |
| 3730 | decl_name, | 3399 | decl_name, |
| 3731 | typed_value.ty, | 3400 | typed_value.ty, |
| 3732 | typed_value.val, | 3401 | typed_value.val, |
| 3733 | required_alignment, | 3402 | required_alignment, |
| 3734 | ); | 3403 | ); |
| 3735 | const addr = try self.allocateAtom(atom, code.len, required_alignment, match); | 3404 | const addr = try self.allocateAtom(atom, code.len, required_alignment, sect_id); |
| 3736 | | 3405 | |
| 3737 | log.debug("allocated atom for {?s} at 0x{x}", .{ name, addr }); | 3406 | log.debug("allocated atom for {?s} at 0x{x}", .{ name, addr }); |
| 3738 | log.debug(" (required alignment 0x{x})", .{required_alignment}); | 3407 | log.debug(" (required alignment 0x{x})", .{required_alignment}); |
| 3739 | | 3408 | |
| 3740 | errdefer self.freeAtom(atom, match, true); | 3409 | errdefer self.freeAtom(atom, sect_id, true); |
| 3741 | | 3410 | |
| 3742 | const symbol = atom.getSymbolPtr(self); | 3411 | const symbol = atom.getSymbolPtr(self); |
| 3743 | symbol.* = .{ | 3412 | symbol.* = .{ |
| 3744 | .n_strx = name_str_index, | 3413 | .n_strx = name_str_index, |
| 3745 | .n_type = macho.N_SECT, | 3414 | .n_type = macho.N_SECT, |
| 3746 | .n_sect = self.getSectionOrdinal(match), | 3415 | .n_sect = sect_id + 1, |
| 3747 | .n_desc = 0, | 3416 | .n_desc = 0, |
| 3748 | .n_value = addr, | 3417 | .n_value = addr, |
| 3749 | }; | 3418 | }; |
| ... | @@ -3894,44 +3563,40 @@ fn needsPointerRebase(ty: Type, val: Value, mod: *Module) bool { | ... | @@ -3894,44 +3563,40 @@ fn needsPointerRebase(ty: Type, val: Value, mod: *Module) bool { |
| 3894 | } | 3563 | } |
| 3895 | } | 3564 | } |
| 3896 | | 3565 | |
| 3897 | fn getMatchingSectionAtom( | 3566 | fn getOutputSectionAtom( |
| 3898 | self: *MachO, | 3567 | self: *MachO, |
| 3899 | atom: *Atom, | 3568 | atom: *Atom, |
| 3900 | name: []const u8, | 3569 | name: []const u8, |
| 3901 | ty: Type, | 3570 | ty: Type, |
| 3902 | val: Value, | 3571 | val: Value, |
| 3903 | alignment: u32, | 3572 | alignment: u32, |
| 3904 | ) !MatchingSection { | 3573 | ) !u8 { |
| 3905 | const code = atom.code.items; | 3574 | const code = atom.code.items; |
| 3906 | const mod = self.base.options.module.?; | 3575 | const mod = self.base.options.module.?; |
| 3907 | const align_log_2 = math.log2(alignment); | 3576 | const align_log_2 = math.log2(alignment); |
| 3908 | const zig_ty = ty.zigTypeTag(); | 3577 | const zig_ty = ty.zigTypeTag(); |
| 3909 | const mode = self.base.options.optimize_mode; | 3578 | const mode = self.base.options.optimize_mode; |
| 3910 | const match: MatchingSection = blk: { | 3579 | const sect_id: u8 = blk: { |
| 3911 | // TODO finish and audit this function | 3580 | // TODO finish and audit this function |
| 3912 | if (val.isUndefDeep()) { | 3581 | if (val.isUndefDeep()) { |
| 3913 | if (mode == .ReleaseFast or mode == .ReleaseSmall) { | 3582 | if (mode == .ReleaseFast or mode == .ReleaseSmall) { |
| 3914 | break :blk MatchingSection{ | 3583 | break :blk (try self.getOutputSection(.{ |
| 3915 | .seg = self.data_segment_cmd_index.?, | 3584 | .segname = makeStaticString("__DATA"), |
| 3916 | .sect = self.bss_section_index.?, | 3585 | .sectname = makeStaticString("__bss"), |
| 3917 | }; | 3586 | .size = code.len, |
| | 3587 | .@"align" = align_log_2, |
| | 3588 | })).?; |
| 3918 | } else { | 3589 | } else { |
| 3919 | break :blk MatchingSection{ | 3590 | break :blk self.data_section_index.?; |
| 3920 | .seg = self.data_segment_cmd_index.?, | | |
| 3921 | .sect = self.data_section_index.?, | | |
| 3922 | }; | | |
| 3923 | } | 3591 | } |
| 3924 | } | 3592 | } |
| 3925 | | 3593 | |
| 3926 | if (val.castTag(.variable)) |_| { | 3594 | if (val.castTag(.variable)) |_| { |
| 3927 | break :blk MatchingSection{ | 3595 | break :blk self.data_section_index.?; |
| 3928 | .seg = self.data_segment_cmd_index.?, | | |
| 3929 | .sect = self.data_section_index.?, | | |
| 3930 | }; | | |
| 3931 | } | 3596 | } |
| 3932 | | 3597 | |
| 3933 | if (needsPointerRebase(ty, val, mod)) { | 3598 | if (needsPointerRebase(ty, val, mod)) { |
| 3934 | break :blk (try self.getMatchingSection(.{ | 3599 | break :blk (try self.getOutputSection(.{ |
| 3935 | .segname = makeStaticString("__DATA_CONST"), | 3600 | .segname = makeStaticString("__DATA_CONST"), |
| 3936 | .sectname = makeStaticString("__const"), | 3601 | .sectname = makeStaticString("__const"), |
| 3937 | .size = code.len, | 3602 | .size = code.len, |
| ... | @@ -3941,10 +3606,7 @@ fn getMatchingSectionAtom( | ... | @@ -3941,10 +3606,7 @@ fn getMatchingSectionAtom( |
| 3941 | | 3606 | |
| 3942 | switch (zig_ty) { | 3607 | switch (zig_ty) { |
| 3943 | .Fn => { | 3608 | .Fn => { |
| 3944 | break :blk MatchingSection{ | 3609 | break :blk self.text_section_index.?; |
| 3945 | .seg = self.text_segment_cmd_index.?, | | |
| 3946 | .sect = self.text_section_index.?, | | |
| 3947 | }; | | |
| 3948 | }, | 3610 | }, |
| 3949 | .Array => { | 3611 | .Array => { |
| 3950 | if (val.tag() == .bytes) { | 3612 | if (val.tag() == .bytes) { |
| ... | @@ -3953,7 +3615,7 @@ fn getMatchingSectionAtom( | ... | @@ -3953,7 +3615,7 @@ fn getMatchingSectionAtom( |
| 3953 | .const_slice_u8_sentinel_0, | 3615 | .const_slice_u8_sentinel_0, |
| 3954 | .manyptr_const_u8_sentinel_0, | 3616 | .manyptr_const_u8_sentinel_0, |
| 3955 | => { | 3617 | => { |
| 3956 | break :blk (try self.getMatchingSection(.{ | 3618 | break :blk (try self.getOutputSection(.{ |
| 3957 | .segname = makeStaticString("__TEXT"), | 3619 | .segname = makeStaticString("__TEXT"), |
| 3958 | .sectname = makeStaticString("__cstring"), | 3620 | .sectname = makeStaticString("__cstring"), |
| 3959 | .flags = macho.S_CSTRING_LITERALS, | 3621 | .flags = macho.S_CSTRING_LITERALS, |
| ... | @@ -3967,22 +3629,21 @@ fn getMatchingSectionAtom( | ... | @@ -3967,22 +3629,21 @@ fn getMatchingSectionAtom( |
| 3967 | }, | 3629 | }, |
| 3968 | else => {}, | 3630 | else => {}, |
| 3969 | } | 3631 | } |
| 3970 | break :blk (try self.getMatchingSection(.{ | 3632 | break :blk (try self.getOutputSection(.{ |
| 3971 | .segname = makeStaticString("__TEXT"), | 3633 | .segname = makeStaticString("__TEXT"), |
| 3972 | .sectname = makeStaticString("__const"), | 3634 | .sectname = makeStaticString("__const"), |
| 3973 | .size = code.len, | 3635 | .size = code.len, |
| 3974 | .@"align" = align_log_2, | 3636 | .@"align" = align_log_2, |
| 3975 | })).?; | 3637 | })).?; |
| 3976 | }; | 3638 | }; |
| 3977 | const sect = self.getSection(match); | 3639 | const header = self.sections.items(.header)[sect_id]; |
| 3978 | log.debug(" allocating atom '{s}' in '{s},{s}' ({d},{d})", .{ | 3640 | log.debug(" allocating atom '{s}' in '{s},{s}', ord({d})", .{ |
| 3979 | name, | 3641 | name, |
| 3980 | sect.segName(), | 3642 | header.segName(), |
| 3981 | sect.sectName(), | 3643 | header.sectName(), |
| 3982 | match.seg, | 3644 | sect_id, |
| 3983 | match.sect, | | |
| 3984 | }); | 3645 | }); |
| 3985 | return match; | 3646 | return sect_id; |
| 3986 | } | 3647 | } |
| 3987 | | 3648 | |
| 3988 | fn placeDecl(self: *MachO, decl_index: Module.Decl.Index, code_len: usize) !u64 { | 3649 | fn placeDecl(self: *MachO, decl_index: Module.Decl.Index, code_len: usize) !u64 { |
| ... | @@ -3996,7 +3657,7 @@ fn placeDecl(self: *MachO, decl_index: Module.Decl.Index, code_len: usize) !u64 | ... | @@ -3996,7 +3657,7 @@ fn placeDecl(self: *MachO, decl_index: Module.Decl.Index, code_len: usize) !u64 |
| 3996 | | 3657 | |
| 3997 | const decl_ptr = self.decls.getPtr(decl_index).?; | 3658 | const decl_ptr = self.decls.getPtr(decl_index).?; |
| 3998 | if (decl_ptr.* == null) { | 3659 | if (decl_ptr.* == null) { |
| 3999 | decl_ptr.* = try self.getMatchingSectionAtom( | 3660 | decl_ptr.* = try self.getOutputSectionAtom( |
| 4000 | &decl.link.macho, | 3661 | &decl.link.macho, |
| 4001 | sym_name, | 3662 | sym_name, |
| 4002 | decl.ty, | 3663 | decl.ty, |
| ... | @@ -4045,7 +3706,7 @@ fn placeDecl(self: *MachO, decl_index: Module.Decl.Index, code_len: usize) !u64 | ... | @@ -4045,7 +3706,7 @@ fn placeDecl(self: *MachO, decl_index: Module.Decl.Index, code_len: usize) !u64 |
| 4045 | symbol.* = .{ | 3706 | symbol.* = .{ |
| 4046 | .n_strx = name_str_index, | 3707 | .n_strx = name_str_index, |
| 4047 | .n_type = macho.N_SECT, | 3708 | .n_type = macho.N_SECT, |
| 4048 | .n_sect = self.getSectionOrdinal(match), | 3709 | .n_sect = match + 1, |
| 4049 | .n_desc = 0, | 3710 | .n_desc = 0, |
| 4050 | .n_value = addr, | 3711 | .n_value = addr, |
| 4051 | }; | 3712 | }; |
| ... | @@ -4134,10 +3795,7 @@ pub fn updateDeclExports( | ... | @@ -4134,10 +3795,7 @@ pub fn updateDeclExports( |
| 4134 | sym.* = .{ | 3795 | sym.* = .{ |
| 4135 | .n_strx = try self.strtab.insert(gpa, exp_name), | 3796 | .n_strx = try self.strtab.insert(gpa, exp_name), |
| 4136 | .n_type = macho.N_SECT | macho.N_EXT, | 3797 | .n_type = macho.N_SECT | macho.N_EXT, |
| 4137 | .n_sect = self.getSectionOrdinal(.{ | 3798 | .n_sect = self.text_section_index.? + 1, // TODO what if we export a variable? |
| 4138 | .seg = self.text_segment_cmd_index.?, | | |
| 4139 | .sect = self.text_section_index.?, // TODO what if we export a variable? | | |
| 4140 | }), | | |
| 4141 | .n_desc = 0, | 3799 | .n_desc = 0, |
| 4142 | .n_value = decl_sym.n_value, | 3800 | .n_value = decl_sym.n_value, |
| 4143 | }; | 3801 | }; |
| ... | @@ -4208,10 +3866,10 @@ pub fn deleteExport(self: *MachO, exp: Export) void { | ... | @@ -4208,10 +3866,10 @@ pub fn deleteExport(self: *MachO, exp: Export) void { |
| 4208 | fn freeUnnamedConsts(self: *MachO, decl_index: Module.Decl.Index) void { | 3866 | fn freeUnnamedConsts(self: *MachO, decl_index: Module.Decl.Index) void { |
| 4209 | const unnamed_consts = self.unnamed_const_atoms.getPtr(decl_index) orelse return; | 3867 | const unnamed_consts = self.unnamed_const_atoms.getPtr(decl_index) orelse return; |
| 4210 | for (unnamed_consts.items) |atom| { | 3868 | for (unnamed_consts.items) |atom| { |
| 4211 | self.freeAtom(atom, .{ | 3869 | // TODO |
| 4212 | .seg = self.text_segment_cmd_index.?, | 3870 | // const sect_id = atom.getSymbol(self).n_sect; |
| 4213 | .sect = self.text_const_section_index.?, | 3871 | const sect_id = self.getSectionByName("__TEXT", "__const").?; |
| 4214 | }, true); | 3872 | self.freeAtom(atom, sect_id, true); |
| 4215 | self.locals_free_list.append(self.base.allocator, atom.sym_index) catch {}; | 3873 | self.locals_free_list.append(self.base.allocator, atom.sym_index) catch {}; |
| 4216 | self.locals.items[atom.sym_index].n_type = 0; | 3874 | self.locals.items[atom.sym_index].n_type = 0; |
| 4217 | _ = self.atom_by_index_table.remove(atom.sym_index); | 3875 | _ = self.atom_by_index_table.remove(atom.sym_index); |
| ... | @@ -4294,6 +3952,7 @@ pub fn getDeclVAddr(self: *MachO, decl_index: Module.Decl.Index, reloc_info: Fil | ... | @@ -4294,6 +3952,7 @@ pub fn getDeclVAddr(self: *MachO, decl_index: Module.Decl.Index, reloc_info: Fil |
| 4294 | } | 3952 | } |
| 4295 | | 3953 | |
| 4296 | fn populateMissingMetadata(self: *MachO) !void { | 3954 | fn populateMissingMetadata(self: *MachO) !void { |
| | 3955 | const gpa = self.base.allocator; |
| 4297 | const cpu_arch = self.base.options.target.cpu.arch; | 3956 | const cpu_arch = self.base.options.target.cpu.arch; |
| 4298 | const pagezero_vmsize = self.base.options.pagezero_size orelse default_pagezero_vmsize; | 3957 | const pagezero_vmsize = self.base.options.pagezero_size orelse default_pagezero_vmsize; |
| 4299 | const aligned_pagezero_vmsize = mem.alignBackwardGeneric(u64, pagezero_vmsize, self.page_size); | 3958 | const aligned_pagezero_vmsize = mem.alignBackwardGeneric(u64, pagezero_vmsize, self.page_size); |
| ... | @@ -4305,21 +3964,16 @@ fn populateMissingMetadata(self: *MachO) !void { | ... | @@ -4305,21 +3964,16 @@ fn populateMissingMetadata(self: *MachO) !void { |
| 4305 | log.warn("requested __PAGEZERO size (0x{x}) is not page aligned", .{pagezero_vmsize}); | 3964 | log.warn("requested __PAGEZERO size (0x{x}) is not page aligned", .{pagezero_vmsize}); |
| 4306 | log.warn(" rounding down to 0x{x}", .{aligned_pagezero_vmsize}); | 3965 | log.warn(" rounding down to 0x{x}", .{aligned_pagezero_vmsize}); |
| 4307 | } | 3966 | } |
| 4308 | self.pagezero_segment_cmd_index = @intCast(u16, self.load_commands.items.len); | 3967 | self.pagezero_segment_cmd_index = @intCast(u8, self.segments.items.len); |
| 4309 | try self.load_commands.append(self.base.allocator, .{ | 3968 | try self.segments.append(gpa, .{ |
| 4310 | .segment = .{ | 3969 | .segname = makeStaticString("__PAGEZERO"), |
| 4311 | .inner = .{ | 3970 | .vmsize = aligned_pagezero_vmsize, |
| 4312 | .segname = makeStaticString("__PAGEZERO"), | 3971 | .cmdsize = @sizeOf(macho.segment_command_64), |
| 4313 | .vmsize = aligned_pagezero_vmsize, | | |
| 4314 | .cmdsize = @sizeOf(macho.segment_command_64), | | |
| 4315 | }, | | |
| 4316 | }, | | |
| 4317 | }); | 3972 | }); |
| 4318 | self.load_commands_dirty = true; | | |
| 4319 | } | 3973 | } |
| 4320 | | 3974 | |
| 4321 | if (self.text_segment_cmd_index == null) { | 3975 | if (self.text_segment_cmd_index == null) { |
| 4322 | self.text_segment_cmd_index = @intCast(u16, self.load_commands.items.len); | 3976 | self.text_segment_cmd_index = @intCast(u8, self.segments.items.len); |
| 4323 | const needed_size = if (self.mode == .incremental) blk: { | 3977 | const needed_size = if (self.mode == .incremental) blk: { |
| 4324 | const headerpad_size = @maximum(self.base.options.headerpad_size orelse 0, default_headerpad_size); | 3978 | const headerpad_size = @maximum(self.base.options.headerpad_size orelse 0, default_headerpad_size); |
| 4325 | const program_code_size_hint = self.base.options.program_code_size_hint; | 3979 | const program_code_size_hint = self.base.options.program_code_size_hint; |
| ... | @@ -4329,20 +3983,15 @@ fn populateMissingMetadata(self: *MachO) !void { | ... | @@ -4329,20 +3983,15 @@ fn populateMissingMetadata(self: *MachO) !void { |
| 4329 | log.debug("found __TEXT segment free space 0x{x} to 0x{x}", .{ 0, needed_size }); | 3983 | log.debug("found __TEXT segment free space 0x{x} to 0x{x}", .{ 0, needed_size }); |
| 4330 | break :blk needed_size; | 3984 | break :blk needed_size; |
| 4331 | } else 0; | 3985 | } else 0; |
| 4332 | try self.load_commands.append(self.base.allocator, .{ | 3986 | try self.segments.append(gpa, .{ |
| 4333 | .segment = .{ | 3987 | .segname = makeStaticString("__TEXT"), |
| 4334 | .inner = .{ | 3988 | .vmaddr = aligned_pagezero_vmsize, |
| 4335 | .segname = makeStaticString("__TEXT"), | 3989 | .vmsize = needed_size, |
| 4336 | .vmaddr = aligned_pagezero_vmsize, | 3990 | .filesize = needed_size, |
| 4337 | .vmsize = needed_size, | 3991 | .maxprot = macho.PROT.READ | macho.PROT.EXEC, |
| 4338 | .filesize = needed_size, | 3992 | .initprot = macho.PROT.READ | macho.PROT.EXEC, |
| 4339 | .maxprot = macho.PROT.READ | macho.PROT.EXEC, | 3993 | .cmdsize = @sizeOf(macho.segment_command_64), |
| 4340 | .initprot = macho.PROT.READ | macho.PROT.EXEC, | | |
| 4341 | .cmdsize = @sizeOf(macho.segment_command_64), | | |
| 4342 | }, | | |
| 4343 | }, | | |
| 4344 | }); | 3994 | }); |
| 4345 | self.load_commands_dirty = true; | | |
| 4346 | } | 3995 | } |
| 4347 | | 3996 | |
| 4348 | if (self.text_section_index == null) { | 3997 | if (self.text_section_index == null) { |
| ... | @@ -4353,7 +4002,7 @@ fn populateMissingMetadata(self: *MachO) !void { | ... | @@ -4353,7 +4002,7 @@ fn populateMissingMetadata(self: *MachO) !void { |
| 4353 | }; | 4002 | }; |
| 4354 | const needed_size = if (self.mode == .incremental) self.base.options.program_code_size_hint else 0; | 4003 | const needed_size = if (self.mode == .incremental) self.base.options.program_code_size_hint else 0; |
| 4355 | self.text_section_index = try self.initSection( | 4004 | self.text_section_index = try self.initSection( |
| 4356 | self.text_segment_cmd_index.?, | 4005 | "__TEXT", |
| 4357 | "__text", | 4006 | "__text", |
| 4358 | needed_size, | 4007 | needed_size, |
| 4359 | alignment, | 4008 | alignment, |
| ... | @@ -4376,7 +4025,7 @@ fn populateMissingMetadata(self: *MachO) !void { | ... | @@ -4376,7 +4025,7 @@ fn populateMissingMetadata(self: *MachO) !void { |
| 4376 | }; | 4025 | }; |
| 4377 | const needed_size = if (self.mode == .incremental) stub_size * self.base.options.symbol_count_hint else 0; | 4026 | const needed_size = if (self.mode == .incremental) stub_size * self.base.options.symbol_count_hint else 0; |
| 4378 | self.stubs_section_index = try self.initSection( | 4027 | self.stubs_section_index = try self.initSection( |
| 4379 | self.text_segment_cmd_index.?, | 4028 | "__TEXT", |
| 4380 | "__stubs", | 4029 | "__stubs", |
| 4381 | needed_size, | 4030 | needed_size, |
| 4382 | alignment, | 4031 | alignment, |
| ... | @@ -4408,7 +4057,7 @@ fn populateMissingMetadata(self: *MachO) !void { | ... | @@ -4408,7 +4057,7 @@ fn populateMissingMetadata(self: *MachO) !void { |
| 4408 | else | 4057 | else |
| 4409 | 0; | 4058 | 0; |
| 4410 | self.stub_helper_section_index = try self.initSection( | 4059 | self.stub_helper_section_index = try self.initSection( |
| 4411 | self.text_segment_cmd_index.?, | 4060 | "__TEXT", |
| 4412 | "__stub_helper", | 4061 | "__stub_helper", |
| 4413 | needed_size, | 4062 | needed_size, |
| 4414 | alignment, | 4063 | alignment, |
| ... | @@ -4419,7 +4068,7 @@ fn populateMissingMetadata(self: *MachO) !void { | ... | @@ -4419,7 +4068,7 @@ fn populateMissingMetadata(self: *MachO) !void { |
| 4419 | } | 4068 | } |
| 4420 | | 4069 | |
| 4421 | if (self.data_const_segment_cmd_index == null) { | 4070 | if (self.data_const_segment_cmd_index == null) { |
| 4422 | self.data_const_segment_cmd_index = @intCast(u16, self.load_commands.items.len); | 4071 | self.data_const_segment_cmd_index = @intCast(u8, self.segments.items.len); |
| 4423 | var vmaddr: u64 = 0; | 4072 | var vmaddr: u64 = 0; |
| 4424 | var fileoff: u64 = 0; | 4073 | var fileoff: u64 = 0; |
| 4425 | var needed_size: u64 = 0; | 4074 | var needed_size: u64 = 0; |
| ... | @@ -4434,21 +4083,16 @@ fn populateMissingMetadata(self: *MachO) !void { | ... | @@ -4434,21 +4083,16 @@ fn populateMissingMetadata(self: *MachO) !void { |
| 4434 | fileoff + needed_size, | 4083 | fileoff + needed_size, |
| 4435 | }); | 4084 | }); |
| 4436 | } | 4085 | } |
| 4437 | try self.load_commands.append(self.base.allocator, .{ | 4086 | try self.segments.append(gpa, .{ |
| 4438 | .segment = .{ | 4087 | .segname = makeStaticString("__DATA_CONST"), |
| 4439 | .inner = .{ | 4088 | .vmaddr = vmaddr, |
| 4440 | .segname = makeStaticString("__DATA_CONST"), | 4089 | .vmsize = needed_size, |
| 4441 | .vmaddr = vmaddr, | 4090 | .fileoff = fileoff, |
| 4442 | .vmsize = needed_size, | 4091 | .filesize = needed_size, |
| 4443 | .fileoff = fileoff, | 4092 | .maxprot = macho.PROT.READ | macho.PROT.WRITE, |
| 4444 | .filesize = needed_size, | 4093 | .initprot = macho.PROT.READ | macho.PROT.WRITE, |
| 4445 | .maxprot = macho.PROT.READ | macho.PROT.WRITE, | 4094 | .cmdsize = @sizeOf(macho.segment_command_64), |
| 4446 | .initprot = macho.PROT.READ | macho.PROT.WRITE, | | |
| 4447 | .cmdsize = @sizeOf(macho.segment_command_64), | | |
| 4448 | }, | | |
| 4449 | }, | | |
| 4450 | }); | 4095 | }); |
| 4451 | self.load_commands_dirty = true; | | |
| 4452 | } | 4096 | } |
| 4453 | | 4097 | |
| 4454 | if (self.got_section_index == null) { | 4098 | if (self.got_section_index == null) { |
| ... | @@ -4458,7 +4102,7 @@ fn populateMissingMetadata(self: *MachO) !void { | ... | @@ -4458,7 +4102,7 @@ fn populateMissingMetadata(self: *MachO) !void { |
| 4458 | 0; | 4102 | 0; |
| 4459 | const alignment: u16 = 3; // 2^3 = @sizeOf(u64) | 4103 | const alignment: u16 = 3; // 2^3 = @sizeOf(u64) |
| 4460 | self.got_section_index = try self.initSection( | 4104 | self.got_section_index = try self.initSection( |
| 4461 | self.data_const_segment_cmd_index.?, | 4105 | "__DATA_CONST", |
| 4462 | "__got", | 4106 | "__got", |
| 4463 | needed_size, | 4107 | needed_size, |
| 4464 | alignment, | 4108 | alignment, |
| ... | @@ -4469,7 +4113,7 @@ fn populateMissingMetadata(self: *MachO) !void { | ... | @@ -4469,7 +4113,7 @@ fn populateMissingMetadata(self: *MachO) !void { |
| 4469 | } | 4113 | } |
| 4470 | | 4114 | |
| 4471 | if (self.data_segment_cmd_index == null) { | 4115 | if (self.data_segment_cmd_index == null) { |
| 4472 | self.data_segment_cmd_index = @intCast(u16, self.load_commands.items.len); | 4116 | self.data_segment_cmd_index = @intCast(u8, self.segments.items.len); |
| 4473 | var vmaddr: u64 = 0; | 4117 | var vmaddr: u64 = 0; |
| 4474 | var fileoff: u64 = 0; | 4118 | var fileoff: u64 = 0; |
| 4475 | var needed_size: u64 = 0; | 4119 | var needed_size: u64 = 0; |
| ... | @@ -4484,21 +4128,16 @@ fn populateMissingMetadata(self: *MachO) !void { | ... | @@ -4484,21 +4128,16 @@ fn populateMissingMetadata(self: *MachO) !void { |
| 4484 | fileoff + needed_size, | 4128 | fileoff + needed_size, |
| 4485 | }); | 4129 | }); |
| 4486 | } | 4130 | } |
| 4487 | try self.load_commands.append(self.base.allocator, .{ | 4131 | try self.segments.append(gpa, .{ |
| 4488 | .segment = .{ | 4132 | .segname = makeStaticString("__DATA"), |
| 4489 | .inner = .{ | 4133 | .vmaddr = vmaddr, |
| 4490 | .segname = makeStaticString("__DATA"), | 4134 | .vmsize = needed_size, |
| 4491 | .vmaddr = vmaddr, | 4135 | .fileoff = fileoff, |
| 4492 | .vmsize = needed_size, | 4136 | .filesize = needed_size, |
| 4493 | .fileoff = fileoff, | 4137 | .maxprot = macho.PROT.READ | macho.PROT.WRITE, |
| 4494 | .filesize = needed_size, | 4138 | .initprot = macho.PROT.READ | macho.PROT.WRITE, |
| 4495 | .maxprot = macho.PROT.READ | macho.PROT.WRITE, | 4139 | .cmdsize = @sizeOf(macho.segment_command_64), |
| 4496 | .initprot = macho.PROT.READ | macho.PROT.WRITE, | | |
| 4497 | .cmdsize = @sizeOf(macho.segment_command_64), | | |
| 4498 | }, | | |
| 4499 | }, | | |
| 4500 | }); | 4140 | }); |
| 4501 | self.load_commands_dirty = true; | | |
| 4502 | } | 4141 | } |
| 4503 | | 4142 | |
| 4504 | if (self.la_symbol_ptr_section_index == null) { | 4143 | if (self.la_symbol_ptr_section_index == null) { |
| ... | @@ -4508,7 +4147,7 @@ fn populateMissingMetadata(self: *MachO) !void { | ... | @@ -4508,7 +4147,7 @@ fn populateMissingMetadata(self: *MachO) !void { |
| 4508 | 0; | 4147 | 0; |
| 4509 | const alignment: u16 = 3; // 2^3 = @sizeOf(u64) | 4148 | const alignment: u16 = 3; // 2^3 = @sizeOf(u64) |
| 4510 | self.la_symbol_ptr_section_index = try self.initSection( | 4149 | self.la_symbol_ptr_section_index = try self.initSection( |
| 4511 | self.data_segment_cmd_index.?, | 4150 | "__DATA", |
| 4512 | "__la_symbol_ptr", | 4151 | "__la_symbol_ptr", |
| 4513 | needed_size, | 4152 | needed_size, |
| 4514 | alignment, | 4153 | alignment, |
| ... | @@ -4525,7 +4164,7 @@ fn populateMissingMetadata(self: *MachO) !void { | ... | @@ -4525,7 +4164,7 @@ fn populateMissingMetadata(self: *MachO) !void { |
| 4525 | 0; | 4164 | 0; |
| 4526 | const alignment: u16 = 3; // 2^3 = @sizeOf(u64) | 4165 | const alignment: u16 = 3; // 2^3 = @sizeOf(u64) |
| 4527 | self.data_section_index = try self.initSection( | 4166 | self.data_section_index = try self.initSection( |
| 4528 | self.data_segment_cmd_index.?, | 4167 | "__DATA", |
| 4529 | "__data", | 4168 | "__data", |
| 4530 | needed_size, | 4169 | needed_size, |
| 4531 | alignment, | 4170 | alignment, |
| ... | @@ -4533,76 +4172,8 @@ fn populateMissingMetadata(self: *MachO) !void { | ... | @@ -4533,76 +4172,8 @@ fn populateMissingMetadata(self: *MachO) !void { |
| 4533 | ); | 4172 | ); |
| 4534 | } | 4173 | } |
| 4535 | | 4174 | |
| 4536 | if (self.tlv_section_index == null) { | | |
| 4537 | const needed_size = if (self.mode == .incremental) | | |
| 4538 | @sizeOf(u64) * self.base.options.symbol_count_hint | | |
| 4539 | else | | |
| 4540 | 0; | | |
| 4541 | const alignment: u16 = 3; // 2^3 = @sizeOf(u64) | | |
| 4542 | self.tlv_section_index = try self.initSection( | | |
| 4543 | self.data_segment_cmd_index.?, | | |
| 4544 | "__thread_vars", | | |
| 4545 | needed_size, | | |
| 4546 | alignment, | | |
| 4547 | .{ | | |
| 4548 | .flags = macho.S_THREAD_LOCAL_VARIABLES, | | |
| 4549 | }, | | |
| 4550 | ); | | |
| 4551 | } | | |
| 4552 | | | |
| 4553 | if (self.tlv_data_section_index == null) { | | |
| 4554 | const needed_size = if (self.mode == .incremental) | | |
| 4555 | @sizeOf(u64) * self.base.options.symbol_count_hint | | |
| 4556 | else | | |
| 4557 | 0; | | |
| 4558 | const alignment: u16 = 3; // 2^3 = @sizeOf(u64) | | |
| 4559 | self.tlv_data_section_index = try self.initSection( | | |
| 4560 | self.data_segment_cmd_index.?, | | |
| 4561 | "__thread_data", | | |
| 4562 | needed_size, | | |
| 4563 | alignment, | | |
| 4564 | .{ | | |
| 4565 | .flags = macho.S_THREAD_LOCAL_REGULAR, | | |
| 4566 | }, | | |
| 4567 | ); | | |
| 4568 | } | | |
| 4569 | | | |
| 4570 | if (self.tlv_bss_section_index == null) { | | |
| 4571 | const needed_size = if (self.mode == .incremental) | | |
| 4572 | @sizeOf(u64) * self.base.options.symbol_count_hint | | |
| 4573 | else | | |
| 4574 | 0; | | |
| 4575 | const alignment: u16 = 3; // 2^3 = @sizeOf(u64) | | |
| 4576 | self.tlv_bss_section_index = try self.initSection( | | |
| 4577 | self.data_segment_cmd_index.?, | | |
| 4578 | "__thread_bss", | | |
| 4579 | needed_size, | | |
| 4580 | alignment, | | |
| 4581 | .{ | | |
| 4582 | .flags = macho.S_THREAD_LOCAL_ZEROFILL, | | |
| 4583 | }, | | |
| 4584 | ); | | |
| 4585 | } | | |
| 4586 | | | |
| 4587 | if (self.bss_section_index == null) { | | |
| 4588 | const needed_size = if (self.mode == .incremental) | | |
| 4589 | @sizeOf(u64) * self.base.options.symbol_count_hint | | |
| 4590 | else | | |
| 4591 | 0; | | |
| 4592 | const alignment: u16 = 3; // 2^3 = @sizeOf(u64) | | |
| 4593 | self.bss_section_index = try self.initSection( | | |
| 4594 | self.data_segment_cmd_index.?, | | |
| 4595 | "__bss", | | |
| 4596 | needed_size, | | |
| 4597 | alignment, | | |
| 4598 | .{ | | |
| 4599 | .flags = macho.S_ZEROFILL, | | |
| 4600 | }, | | |
| 4601 | ); | | |
| 4602 | } | | |
| 4603 | | | |
| 4604 | if (self.linkedit_segment_cmd_index == null) { | 4175 | if (self.linkedit_segment_cmd_index == null) { |
| 4605 | self.linkedit_segment_cmd_index = @intCast(u16, self.load_commands.items.len); | 4176 | self.linkedit_segment_cmd_index = @intCast(u8, self.segments.items.len); |
| 4606 | var vmaddr: u64 = 0; | 4177 | var vmaddr: u64 = 0; |
| 4607 | var fileoff: u64 = 0; | 4178 | var fileoff: u64 = 0; |
| 4608 | if (self.mode == .incremental) { | 4179 | if (self.mode == .incremental) { |
| ... | @@ -4611,249 +4182,113 @@ fn populateMissingMetadata(self: *MachO) !void { | ... | @@ -4611,249 +4182,113 @@ fn populateMissingMetadata(self: *MachO) !void { |
| 4611 | fileoff = base.fileoff; | 4182 | fileoff = base.fileoff; |
| 4612 | log.debug("found __LINKEDIT segment free space at 0x{x}", .{fileoff}); | 4183 | log.debug("found __LINKEDIT segment free space at 0x{x}", .{fileoff}); |
| 4613 | } | 4184 | } |
| 4614 | try self.load_commands.append(self.base.allocator, .{ | 4185 | try self.segments.append(gpa, .{ |
| 4615 | .segment = .{ | 4186 | .segname = makeStaticString("__LINKEDIT"), |
| 4616 | .inner = .{ | 4187 | .vmaddr = vmaddr, |
| 4617 | .segname = makeStaticString("__LINKEDIT"), | 4188 | .fileoff = fileoff, |
| 4618 | .vmaddr = vmaddr, | 4189 | .maxprot = macho.PROT.READ, |
| 4619 | .fileoff = fileoff, | 4190 | .initprot = macho.PROT.READ, |
| 4620 | .maxprot = macho.PROT.READ, | 4191 | .cmdsize = @sizeOf(macho.segment_command_64), |
| 4621 | .initprot = macho.PROT.READ, | | |
| 4622 | .cmdsize = @sizeOf(macho.segment_command_64), | | |
| 4623 | }, | | |
| 4624 | }, | | |
| 4625 | }); | | |
| 4626 | self.load_commands_dirty = true; | | |
| 4627 | } | | |
| 4628 | | | |
| 4629 | if (self.dyld_info_cmd_index == null) { | | |
| 4630 | self.dyld_info_cmd_index = @intCast(u16, self.load_commands.items.len); | | |
| 4631 | try self.load_commands.append(self.base.allocator, .{ | | |
| 4632 | .dyld_info_only = .{ | | |
| 4633 | .cmd = .DYLD_INFO_ONLY, | | |
| 4634 | .cmdsize = @sizeOf(macho.dyld_info_command), | | |
| 4635 | .rebase_off = 0, | | |
| 4636 | .rebase_size = 0, | | |
| 4637 | .bind_off = 0, | | |
| 4638 | .bind_size = 0, | | |
| 4639 | .weak_bind_off = 0, | | |
| 4640 | .weak_bind_size = 0, | | |
| 4641 | .lazy_bind_off = 0, | | |
| 4642 | .lazy_bind_size = 0, | | |
| 4643 | .export_off = 0, | | |
| 4644 | .export_size = 0, | | |
| 4645 | }, | | |
| 4646 | }); | 4192 | }); |
| 4647 | self.load_commands_dirty = true; | | |
| 4648 | } | | |
| 4649 | | | |
| 4650 | if (self.symtab_cmd_index == null) { | | |
| 4651 | self.symtab_cmd_index = @intCast(u16, self.load_commands.items.len); | | |
| 4652 | try self.load_commands.append(self.base.allocator, .{ | | |
| 4653 | .symtab = .{ | | |
| 4654 | .cmdsize = @sizeOf(macho.symtab_command), | | |
| 4655 | .symoff = 0, | | |
| 4656 | .nsyms = 0, | | |
| 4657 | .stroff = 0, | | |
| 4658 | .strsize = 0, | | |
| 4659 | }, | | |
| 4660 | }); | | |
| 4661 | self.load_commands_dirty = true; | | |
| 4662 | } | | |
| 4663 | | | |
| 4664 | if (self.dysymtab_cmd_index == null) { | | |
| 4665 | self.dysymtab_cmd_index = @intCast(u16, self.load_commands.items.len); | | |
| 4666 | try self.load_commands.append(self.base.allocator, .{ | | |
| 4667 | .dysymtab = .{ | | |
| 4668 | .cmdsize = @sizeOf(macho.dysymtab_command), | | |
| 4669 | .ilocalsym = 0, | | |
| 4670 | .nlocalsym = 0, | | |
| 4671 | .iextdefsym = 0, | | |
| 4672 | .nextdefsym = 0, | | |
| 4673 | .iundefsym = 0, | | |
| 4674 | .nundefsym = 0, | | |
| 4675 | .tocoff = 0, | | |
| 4676 | .ntoc = 0, | | |
| 4677 | .modtaboff = 0, | | |
| 4678 | .nmodtab = 0, | | |
| 4679 | .extrefsymoff = 0, | | |
| 4680 | .nextrefsyms = 0, | | |
| 4681 | .indirectsymoff = 0, | | |
| 4682 | .nindirectsyms = 0, | | |
| 4683 | .extreloff = 0, | | |
| 4684 | .nextrel = 0, | | |
| 4685 | .locreloff = 0, | | |
| 4686 | .nlocrel = 0, | | |
| 4687 | }, | | |
| 4688 | }); | | |
| 4689 | self.load_commands_dirty = true; | | |
| 4690 | } | 4193 | } |
| | 4194 | } |
| 4691 | | 4195 | |
| 4692 | if (self.dylinker_cmd_index == null) { | 4196 | inline fn calcInstallNameLen(cmd_size: u64, name: []const u8, assume_max_path_len: bool) u64 { |
| 4693 | self.dylinker_cmd_index = @intCast(u16, self.load_commands.items.len); | 4197 | const name_len = if (assume_max_path_len) std.os.PATH_MAX else std.mem.len(name) + 1; |
| 4694 | const cmdsize = @intCast(u32, mem.alignForwardGeneric( | 4198 | return mem.alignForwardGeneric(u64, cmd_size + name_len, @alignOf(u64)); |
| 4695 | u64, | 4199 | } |
| 4696 | @sizeOf(macho.dylinker_command) + mem.sliceTo(default_dyld_path, 0).len, | | |
| 4697 | @sizeOf(u64), | | |
| 4698 | )); | | |
| 4699 | var dylinker_cmd = macho.emptyGenericCommandWithData(macho.dylinker_command{ | | |
| 4700 | .cmd = .LOAD_DYLINKER, | | |
| 4701 | .cmdsize = cmdsize, | | |
| 4702 | .name = @sizeOf(macho.dylinker_command), | | |
| 4703 | }); | | |
| 4704 | dylinker_cmd.data = try self.base.allocator.alloc(u8, cmdsize - dylinker_cmd.inner.name); | | |
| 4705 | mem.set(u8, dylinker_cmd.data, 0); | | |
| 4706 | mem.copy(u8, dylinker_cmd.data, mem.sliceTo(default_dyld_path, 0)); | | |
| 4707 | try self.load_commands.append(self.base.allocator, .{ .dylinker = dylinker_cmd }); | | |
| 4708 | self.load_commands_dirty = true; | | |
| 4709 | } | | |
| 4710 | | | |
| 4711 | if (self.main_cmd_index == null and self.base.options.output_mode == .Exe) { | | |
| 4712 | self.main_cmd_index = @intCast(u16, self.load_commands.items.len); | | |
| 4713 | try self.load_commands.append(self.base.allocator, .{ | | |
| 4714 | .main = .{ | | |
| 4715 | .cmdsize = @sizeOf(macho.entry_point_command), | | |
| 4716 | .entryoff = 0x0, | | |
| 4717 | .stacksize = 0, | | |
| 4718 | }, | | |
| 4719 | }); | | |
| 4720 | self.load_commands_dirty = true; | | |
| 4721 | } | | |
| 4722 | | 4200 | |
| 4723 | if (self.dylib_id_cmd_index == null and self.base.options.output_mode == .Lib) { | 4201 | fn calcLCsSize(self: *MachO, assume_max_path_len: bool) !u32 { |
| 4724 | self.dylib_id_cmd_index = @intCast(u16, self.load_commands.items.len); | 4202 | const gpa = self.base.allocator; |
| 4725 | const install_name = self.base.options.install_name orelse self.base.options.emit.?.sub_path; | 4203 | var sizeofcmds: u64 = 0; |
| 4726 | const current_version = self.base.options.version orelse | 4204 | for (self.segments.items) |seg| { |
| 4727 | std.builtin.Version{ .major = 1, .minor = 0, .patch = 0 }; | 4205 | sizeofcmds += seg.nsects * @sizeOf(macho.section_64) + @sizeOf(macho.segment_command_64); |
| 4728 | const compat_version = self.base.options.compatibility_version orelse | 4206 | } |
| 4729 | std.builtin.Version{ .major = 1, .minor = 0, .patch = 0 }; | 4207 | |
| 4730 | var dylib_cmd = try macho.createLoadDylibCommand( | 4208 | // LC_DYLD_INFO_ONLY |
| 4731 | self.base.allocator, | 4209 | sizeofcmds += @sizeOf(macho.dyld_info_command); |
| 4732 | .ID_DYLIB, | 4210 | // LC_FUNCTION_STARTS |
| 4733 | install_name, | 4211 | if (self.text_section_index != null) { |
| 4734 | 2, | 4212 | sizeofcmds += @sizeOf(macho.linkedit_data_command); |
| 4735 | current_version.major << 16 | current_version.minor << 8 | current_version.patch, | 4213 | } |
| 4736 | compat_version.major << 16 | compat_version.minor << 8 | compat_version.patch, | 4214 | // LC_DATA_IN_CODE |
| 4737 | ); | 4215 | sizeofcmds += @sizeOf(macho.linkedit_data_command); |
| 4738 | errdefer dylib_cmd.deinit(self.base.allocator); | 4216 | // LC_SYMTAB |
| 4739 | try self.load_commands.append(self.base.allocator, .{ .dylib = dylib_cmd }); | 4217 | sizeofcmds += @sizeOf(macho.symtab_command); |
| 4740 | self.load_commands_dirty = true; | 4218 | // LC_DYSYMTAB |
| | 4219 | sizeofcmds += @sizeOf(macho.dysymtab_command); |
| | 4220 | // LC_LOAD_DYLINKER |
| | 4221 | sizeofcmds += calcInstallNameLen( |
| | 4222 | @sizeOf(macho.dylinker_command), |
| | 4223 | mem.sliceTo(default_dyld_path, 0), |
| | 4224 | false, |
| | 4225 | ); |
| | 4226 | // LC_MAIN |
| | 4227 | if (self.base.options.output_mode == .Exe) { |
| | 4228 | sizeofcmds += @sizeOf(macho.entry_point_command); |
| | 4229 | } |
| | 4230 | // LC_ID_DYLIB |
| | 4231 | if (self.base.options.output_mode == .Lib) { |
| | 4232 | sizeofcmds += blk: { |
| | 4233 | const install_name = self.base.options.install_name orelse self.base.options.emit.?.sub_path; |
| | 4234 | break :blk calcInstallNameLen( |
| | 4235 | @sizeOf(macho.dylib_command), |
| | 4236 | install_name, |
| | 4237 | assume_max_path_len, |
| | 4238 | ); |
| | 4239 | }; |
| 4741 | } | 4240 | } |
| 4742 | | 4241 | // LC_RPATH |
| 4743 | if (self.source_version_cmd_index == null) { | 4242 | { |
| 4744 | self.source_version_cmd_index = @intCast(u16, self.load_commands.items.len); | 4243 | var it = RpathIterator.init(gpa, self.base.options.rpath_list); |
| 4745 | try self.load_commands.append(self.base.allocator, .{ | 4244 | defer it.deinit(); |
| 4746 | .source_version = .{ | 4245 | while (try it.next()) |rpath| { |
| 4747 | .cmdsize = @sizeOf(macho.source_version_command), | 4246 | sizeofcmds += calcInstallNameLen( |
| 4748 | .version = 0x0, | 4247 | @sizeOf(macho.rpath_command), |
| 4749 | }, | 4248 | rpath, |
| 4750 | }); | 4249 | assume_max_path_len, |
| 4751 | self.load_commands_dirty = true; | 4250 | ); |
| | 4251 | } |
| 4752 | } | 4252 | } |
| 4753 | | 4253 | // LC_SOURCE_VERSION |
| 4754 | if (self.build_version_cmd_index == null) { | 4254 | sizeofcmds += @sizeOf(macho.source_version_command); |
| 4755 | self.build_version_cmd_index = @intCast(u16, self.load_commands.items.len); | 4255 | // LC_BUILD_VERSION |
| 4756 | const cmdsize = @intCast(u32, mem.alignForwardGeneric( | 4256 | sizeofcmds += @sizeOf(macho.build_version_command) + @sizeOf(macho.build_tool_version); |
| 4757 | u64, | 4257 | // LC_UUID |
| 4758 | @sizeOf(macho.build_version_command) + @sizeOf(macho.build_tool_version), | 4258 | sizeofcmds += @sizeOf(macho.uuid_command); |
| 4759 | @sizeOf(u64), | 4259 | // LC_LOAD_DYLIB |
| 4760 | )); | 4260 | for (self.referenced_dylibs.keys()) |id| { |
| 4761 | const platform_version = blk: { | 4261 | const dylib = self.dylibs.items[id]; |
| 4762 | const ver = self.base.options.target.os.version_range.semver.min; | 4262 | const dylib_id = dylib.id orelse unreachable; |
| 4763 | const platform_version = ver.major << 16 | ver.minor << 8; | 4263 | sizeofcmds += calcInstallNameLen( |
| 4764 | break :blk platform_version; | 4264 | @sizeOf(macho.dylib_command), |
| 4765 | }; | 4265 | dylib_id.name, |
| 4766 | const sdk_version = if (self.base.options.native_darwin_sdk) |sdk| blk: { | 4266 | assume_max_path_len, |
| 4767 | const ver = sdk.version; | 4267 | ); |
| 4768 | const sdk_version = ver.major << 16 | ver.minor << 8; | 4268 | } |
| 4769 | break :blk sdk_version; | 4269 | // LC_CODE_SIGNATURE |
| 4770 | } else platform_version; | 4270 | { |
| 4771 | const is_simulator_abi = self.base.options.target.abi == .simulator; | 4271 | const target = self.base.options.target; |
| 4772 | var cmd = macho.emptyGenericCommandWithData(macho.build_version_command{ | 4272 | const requires_codesig = blk: { |
| 4773 | .cmdsize = cmdsize, | 4273 | if (self.base.options.entitlements) |_| break :blk true; |
| 4774 | .platform = switch (self.base.options.target.os.tag) { | 4274 | if (target.cpu.arch == .aarch64 and (target.os.tag == .macos or target.abi == .simulator)) |
| 4775 | .macos => .MACOS, | 4275 | break :blk true; |
| 4776 | .ios => if (is_simulator_abi) macho.PLATFORM.IOSSIMULATOR else macho.PLATFORM.IOS, | 4276 | break :blk false; |
| 4777 | .watchos => if (is_simulator_abi) macho.PLATFORM.WATCHOSSIMULATOR else macho.PLATFORM.WATCHOS, | | |
| 4778 | .tvos => if (is_simulator_abi) macho.PLATFORM.TVOSSIMULATOR else macho.PLATFORM.TVOS, | | |
| 4779 | else => unreachable, | | |
| 4780 | }, | | |
| 4781 | .minos = platform_version, | | |
| 4782 | .sdk = sdk_version, | | |
| 4783 | .ntools = 1, | | |
| 4784 | }); | | |
| 4785 | const ld_ver = macho.build_tool_version{ | | |
| 4786 | .tool = .LD, | | |
| 4787 | .version = 0x0, | | |
| 4788 | }; | | |
| 4789 | cmd.data = try self.base.allocator.alloc(u8, cmdsize - @sizeOf(macho.build_version_command)); | | |
| 4790 | mem.set(u8, cmd.data, 0); | | |
| 4791 | mem.copy(u8, cmd.data, mem.asBytes(&ld_ver)); | | |
| 4792 | try self.load_commands.append(self.base.allocator, .{ .build_version = cmd }); | | |
| 4793 | self.load_commands_dirty = true; | | |
| 4794 | } | | |
| 4795 | | | |
| 4796 | if (self.uuid_cmd_index == null) { | | |
| 4797 | self.uuid_cmd_index = @intCast(u16, self.load_commands.items.len); | | |
| 4798 | var uuid_cmd: macho.uuid_command = .{ | | |
| 4799 | .cmdsize = @sizeOf(macho.uuid_command), | | |
| 4800 | .uuid = undefined, | | |
| 4801 | }; | 4277 | }; |
| 4802 | std.crypto.random.bytes(&uuid_cmd.uuid); | 4278 | if (requires_codesig) { |
| 4803 | try self.load_commands.append(self.base.allocator, .{ .uuid = uuid_cmd }); | 4279 | sizeofcmds += @sizeOf(macho.linkedit_data_command); |
| 4804 | self.load_commands_dirty = true; | 4280 | } |
| 4805 | } | | |
| 4806 | | | |
| 4807 | if (self.function_starts_cmd_index == null) { | | |
| 4808 | self.function_starts_cmd_index = @intCast(u16, self.load_commands.items.len); | | |
| 4809 | try self.load_commands.append(self.base.allocator, .{ | | |
| 4810 | .linkedit_data = .{ | | |
| 4811 | .cmd = .FUNCTION_STARTS, | | |
| 4812 | .cmdsize = @sizeOf(macho.linkedit_data_command), | | |
| 4813 | .dataoff = 0, | | |
| 4814 | .datasize = 0, | | |
| 4815 | }, | | |
| 4816 | }); | | |
| 4817 | self.load_commands_dirty = true; | | |
| 4818 | } | 4281 | } |
| 4819 | | 4282 | |
| 4820 | if (self.data_in_code_cmd_index == null) { | 4283 | return @intCast(u32, sizeofcmds); |
| 4821 | self.data_in_code_cmd_index = @intCast(u16, self.load_commands.items.len); | | |
| 4822 | try self.load_commands.append(self.base.allocator, .{ | | |
| 4823 | .linkedit_data = .{ | | |
| 4824 | .cmd = .DATA_IN_CODE, | | |
| 4825 | .cmdsize = @sizeOf(macho.linkedit_data_command), | | |
| 4826 | .dataoff = 0, | | |
| 4827 | .datasize = 0, | | |
| 4828 | }, | | |
| 4829 | }); | | |
| 4830 | self.load_commands_dirty = true; | | |
| 4831 | } | | |
| 4832 | } | 4284 | } |
| 4833 | | 4285 | |
| 4834 | fn calcMinHeaderpad(self: *MachO) u64 { | 4286 | fn calcMinHeaderPad(self: *MachO) !u64 { |
| 4835 | var sizeofcmds: u32 = 0; | 4287 | var padding: u32 = (try self.calcLCsSize(false)) + (self.base.options.headerpad_size orelse 0); |
| 4836 | for (self.load_commands.items) |lc| { | | |
| 4837 | if (lc.cmd() == .NONE) continue; | | |
| 4838 | sizeofcmds += lc.cmdsize(); | | |
| 4839 | } | | |
| 4840 | | | |
| 4841 | var padding: u32 = sizeofcmds + (self.base.options.headerpad_size orelse 0); | | |
| 4842 | log.debug("minimum requested headerpad size 0x{x}", .{padding + @sizeOf(macho.mach_header_64)}); | 4288 | log.debug("minimum requested headerpad size 0x{x}", .{padding + @sizeOf(macho.mach_header_64)}); |
| 4843 | | 4289 | |
| 4844 | if (self.base.options.headerpad_max_install_names) { | 4290 | if (self.base.options.headerpad_max_install_names) { |
| 4845 | var min_headerpad_size: u32 = 0; | 4291 | var min_headerpad_size: u32 = try self.calcLCsSize(true); |
| 4846 | for (self.load_commands.items) |lc| switch (lc.cmd()) { | | |
| 4847 | .ID_DYLIB, | | |
| 4848 | .LOAD_WEAK_DYLIB, | | |
| 4849 | .LOAD_DYLIB, | | |
| 4850 | .REEXPORT_DYLIB, | | |
| 4851 | => { | | |
| 4852 | min_headerpad_size += @sizeOf(macho.dylib_command) + std.os.PATH_MAX + 1; | | |
| 4853 | }, | | |
| 4854 | | | |
| 4855 | else => {}, | | |
| 4856 | }; | | |
| 4857 | log.debug("headerpad_max_install_names minimum headerpad size 0x{x}", .{ | 4292 | log.debug("headerpad_max_install_names minimum headerpad size 0x{x}", .{ |
| 4858 | min_headerpad_size + @sizeOf(macho.mach_header_64), | 4293 | min_headerpad_size + @sizeOf(macho.mach_header_64), |
| 4859 | }); | 4294 | }); |
| ... | @@ -4868,32 +4303,32 @@ fn calcMinHeaderpad(self: *MachO) u64 { | ... | @@ -4868,32 +4303,32 @@ fn calcMinHeaderpad(self: *MachO) u64 { |
| 4868 | fn allocateSegments(self: *MachO) !void { | 4303 | fn allocateSegments(self: *MachO) !void { |
| 4869 | try self.allocateSegment(self.text_segment_cmd_index, &.{ | 4304 | try self.allocateSegment(self.text_segment_cmd_index, &.{ |
| 4870 | self.pagezero_segment_cmd_index, | 4305 | self.pagezero_segment_cmd_index, |
| 4871 | }, self.calcMinHeaderpad()); | 4306 | }, try self.calcMinHeaderPad()); |
| 4872 | | 4307 | |
| 4873 | if (self.text_segment_cmd_index) |index| blk: { | 4308 | if (self.text_segment_cmd_index) |index| blk: { |
| 4874 | const seg = &self.load_commands.items[index].segment; | 4309 | const indexes = self.getSectionIndexes(index); |
| 4875 | if (seg.sections.items.len == 0) break :blk; | 4310 | if (indexes.start == indexes.end) break :blk; |
| | 4311 | const seg = self.segments.items[index]; |
| 4876 | | 4312 | |
| 4877 | // Shift all sections to the back to minimize jump size between __TEXT and __DATA segments. | 4313 | // Shift all sections to the back to minimize jump size between __TEXT and __DATA segments. |
| 4878 | var min_alignment: u32 = 0; | 4314 | var min_alignment: u32 = 0; |
| 4879 | for (seg.sections.items) |sect| { | 4315 | for (self.sections.items(.header)[indexes.start..indexes.end]) |header| { |
| 4880 | const alignment = try math.powi(u32, 2, sect.@"align"); | 4316 | const alignment = try math.powi(u32, 2, header.@"align"); |
| 4881 | min_alignment = math.max(min_alignment, alignment); | 4317 | min_alignment = math.max(min_alignment, alignment); |
| 4882 | } | 4318 | } |
| 4883 | | 4319 | |
| 4884 | assert(min_alignment > 0); | 4320 | assert(min_alignment > 0); |
| 4885 | const last_sect_idx = seg.sections.items.len - 1; | 4321 | const last_header = self.sections.items(.header)[indexes.end - 1]; |
| 4886 | const last_sect = seg.sections.items[last_sect_idx]; | | |
| 4887 | const shift: u32 = shift: { | 4322 | const shift: u32 = shift: { |
| 4888 | const diff = seg.inner.filesize - last_sect.offset - last_sect.size; | 4323 | const diff = seg.filesize - last_header.offset - last_header.size; |
| 4889 | const factor = @divTrunc(diff, min_alignment); | 4324 | const factor = @divTrunc(diff, min_alignment); |
| 4890 | break :shift @intCast(u32, factor * min_alignment); | 4325 | break :shift @intCast(u32, factor * min_alignment); |
| 4891 | }; | 4326 | }; |
| 4892 | | 4327 | |
| 4893 | if (shift > 0) { | 4328 | if (shift > 0) { |
| 4894 | for (seg.sections.items) |*sect| { | 4329 | for (self.sections.items(.header)[indexes.start..indexes.end]) |*header| { |
| 4895 | sect.offset += shift; | 4330 | header.offset += shift; |
| 4896 | sect.addr += shift; | 4331 | header.addr += shift; |
| 4897 | } | 4332 | } |
| 4898 | } | 4333 | } |
| 4899 | } | 4334 | } |
| ... | @@ -4917,42 +4352,40 @@ fn allocateSegments(self: *MachO) !void { | ... | @@ -4917,42 +4352,40 @@ fn allocateSegments(self: *MachO) !void { |
| 4917 | }, 0); | 4352 | }, 0); |
| 4918 | } | 4353 | } |
| 4919 | | 4354 | |
| 4920 | fn allocateSegment(self: *MachO, maybe_index: ?u16, indices: []const ?u16, init_size: u64) !void { | 4355 | fn allocateSegment(self: *MachO, maybe_index: ?u8, indices: []const ?u8, init_size: u64) !void { |
| 4921 | const index = maybe_index orelse return; | 4356 | const index = maybe_index orelse return; |
| 4922 | const seg = &self.load_commands.items[index].segment; | 4357 | const seg = &self.segments.items[index]; |
| 4923 | | 4358 | |
| 4924 | const base = self.getSegmentAllocBase(indices); | 4359 | const base = self.getSegmentAllocBase(indices); |
| 4925 | seg.inner.vmaddr = base.vmaddr; | 4360 | seg.vmaddr = base.vmaddr; |
| 4926 | seg.inner.fileoff = base.fileoff; | 4361 | seg.fileoff = base.fileoff; |
| 4927 | seg.inner.filesize = init_size; | 4362 | seg.filesize = init_size; |
| 4928 | seg.inner.vmsize = init_size; | 4363 | seg.vmsize = init_size; |
| 4929 | | 4364 | |
| 4930 | // Allocate the sections according to their alignment at the beginning of the segment. | 4365 | // Allocate the sections according to their alignment at the beginning of the segment. |
| | 4366 | const indexes = self.getSectionIndexes(index); |
| 4931 | var start = init_size; | 4367 | var start = init_size; |
| 4932 | for (seg.sections.items) |*sect| { | 4368 | const slice = self.sections.slice(); |
| 4933 | const is_zerofill = sect.flags == macho.S_ZEROFILL or sect.flags == macho.S_THREAD_LOCAL_ZEROFILL; | 4369 | for (slice.items(.header)[indexes.start..indexes.end]) |*header| { |
| 4934 | const use_llvm = build_options.have_llvm and self.base.options.use_llvm; | 4370 | const alignment = try math.powi(u32, 2, header.@"align"); |
| 4935 | const use_stage1 = build_options.is_stage1 and self.base.options.use_stage1; | | |
| 4936 | const alignment = try math.powi(u32, 2, sect.@"align"); | | |
| 4937 | const start_aligned = mem.alignForwardGeneric(u64, start, alignment); | 4371 | const start_aligned = mem.alignForwardGeneric(u64, start, alignment); |
| 4938 | | 4372 | |
| 4939 | // TODO handle zerofill sections in stage2 | 4373 | header.offset = if (header.isZerofill()) |
| 4940 | sect.offset = if (is_zerofill and (use_stage1 or use_llvm)) | | |
| 4941 | 0 | 4374 | 0 |
| 4942 | else | 4375 | else |
| 4943 | @intCast(u32, seg.inner.fileoff + start_aligned); | 4376 | @intCast(u32, seg.fileoff + start_aligned); |
| 4944 | sect.addr = seg.inner.vmaddr + start_aligned; | 4377 | header.addr = seg.vmaddr + start_aligned; |
| 4945 | | 4378 | |
| 4946 | start = start_aligned + sect.size; | 4379 | start = start_aligned + header.size; |
| 4947 | | 4380 | |
| 4948 | if (!(is_zerofill and (use_stage1 or use_llvm))) { | 4381 | if (!header.isZerofill()) { |
| 4949 | seg.inner.filesize = start; | 4382 | seg.filesize = start; |
| 4950 | } | 4383 | } |
| 4951 | seg.inner.vmsize = start; | 4384 | seg.vmsize = start; |
| 4952 | } | 4385 | } |
| 4953 | | 4386 | |
| 4954 | seg.inner.filesize = mem.alignForwardGeneric(u64, seg.inner.filesize, self.page_size); | 4387 | seg.filesize = mem.alignForwardGeneric(u64, seg.filesize, self.page_size); |
| 4955 | seg.inner.vmsize = mem.alignForwardGeneric(u64, seg.inner.vmsize, self.page_size); | 4388 | seg.vmsize = mem.alignForwardGeneric(u64, seg.vmsize, self.page_size); |
| 4956 | } | 4389 | } |
| 4957 | | 4390 | |
| 4958 | const InitSectionOpts = struct { | 4391 | const InitSectionOpts = struct { |
| ... | @@ -4963,184 +4396,270 @@ const InitSectionOpts = struct { | ... | @@ -4963,184 +4396,270 @@ const InitSectionOpts = struct { |
| 4963 | | 4396 | |
| 4964 | fn initSection( | 4397 | fn initSection( |
| 4965 | self: *MachO, | 4398 | self: *MachO, |
| 4966 | segment_id: u16, | 4399 | segname: []const u8, |
| 4967 | sectname: []const u8, | 4400 | sectname: []const u8, |
| 4968 | size: u64, | 4401 | size: u64, |
| 4969 | alignment: u32, | 4402 | alignment: u32, |
| 4970 | opts: InitSectionOpts, | 4403 | opts: InitSectionOpts, |
| 4971 | ) !u16 { | 4404 | ) !u8 { |
| 4972 | const seg = &self.load_commands.items[segment_id].segment; | 4405 | const segment_id = self.getSegmentByName(segname).?; |
| 4973 | var sect = macho.section_64{ | 4406 | const seg = &self.segments.items[segment_id]; |
| | 4407 | const index = try self.insertSection(segment_id, .{ |
| 4974 | .sectname = makeStaticString(sectname), | 4408 | .sectname = makeStaticString(sectname), |
| 4975 | .segname = seg.inner.segname, | 4409 | .segname = seg.segname, |
| 4976 | .size = if (self.mode == .incremental) @intCast(u32, size) else 0, | | |
| 4977 | .@"align" = alignment, | | |
| 4978 | .flags = opts.flags, | 4410 | .flags = opts.flags, |
| 4979 | .reserved1 = opts.reserved1, | 4411 | .reserved1 = opts.reserved1, |
| 4980 | .reserved2 = opts.reserved2, | 4412 | .reserved2 = opts.reserved2, |
| 4981 | }; | 4413 | }); |
| | 4414 | seg.cmdsize += @sizeOf(macho.section_64); |
| | 4415 | seg.nsects += 1; |
| 4982 | | 4416 | |
| 4983 | if (self.mode == .incremental) { | 4417 | if (self.mode == .incremental) { |
| | 4418 | const header = &self.sections.items(.header)[index]; |
| | 4419 | header.size = size; |
| | 4420 | header.@"align" = alignment; |
| | 4421 | |
| | 4422 | const prev_end_off = if (index > 0) blk: { |
| | 4423 | const prev_section = self.sections.get(index - 1); |
| | 4424 | if (prev_section.segment_index == segment_id) { |
| | 4425 | const prev_header = prev_section.header; |
| | 4426 | break :blk prev_header.offset + padToIdeal(prev_header.size); |
| | 4427 | } else break :blk seg.fileoff; |
| | 4428 | } else 0; |
| 4984 | const alignment_pow_2 = try math.powi(u32, 2, alignment); | 4429 | const alignment_pow_2 = try math.powi(u32, 2, alignment); |
| 4985 | const padding: ?u32 = if (segment_id == self.text_segment_cmd_index.?) | 4430 | // TODO better prealloc for __text section |
| 4986 | @maximum(self.base.options.headerpad_size orelse 0, default_headerpad_size) | 4431 | // const padding: u64 = if (index == 0) try self.calcMinHeaderPad() else 0; |
| 4987 | else | 4432 | const padding: u64 = if (index == 0) 0x1000 else 0; |
| 4988 | null; | 4433 | const off = mem.alignForwardGeneric(u64, padding + prev_end_off, alignment_pow_2); |
| 4989 | const off = self.findFreeSpace(segment_id, alignment_pow_2, padding); | | |
| 4990 | log.debug("allocating {s},{s} section from 0x{x} to 0x{x}", .{ | | |
| 4991 | sect.segName(), | | |
| 4992 | sect.sectName(), | | |
| 4993 | off, | | |
| 4994 | off + size, | | |
| 4995 | }); | | |
| 4996 | | 4434 | |
| 4997 | sect.addr = seg.inner.vmaddr + off - seg.inner.fileoff; | 4435 | if (!header.isZerofill()) { |
| | 4436 | header.offset = @intCast(u32, off); |
| | 4437 | } |
| | 4438 | header.addr = seg.vmaddr + off - seg.fileoff; |
| 4998 | | 4439 | |
| 4999 | const is_zerofill = opts.flags == macho.S_ZEROFILL or opts.flags == macho.S_THREAD_LOCAL_ZEROFILL; | 4440 | // TODO Will this break if we are inserting section that is not the last section |
| 5000 | const use_llvm = build_options.have_llvm and self.base.options.use_llvm; | 4441 | // in a segment? |
| 5001 | const use_stage1 = build_options.is_stage1 and self.base.options.use_stage1; | 4442 | const max_size = self.allocatedSize(segment_id, off); |
| 5002 | | 4443 | |
| 5003 | // TODO handle zerofill in stage2 | 4444 | if (size > max_size) { |
| 5004 | if (!(is_zerofill and (use_stage1 or use_llvm))) { | 4445 | try self.growSection(index, @intCast(u32, size)); |
| 5005 | sect.offset = @intCast(u32, off); | | |
| 5006 | } | 4446 | } |
| | 4447 | |
| | 4448 | log.debug("allocating {s},{s} section at 0x{x}", .{ header.segName(), header.sectName(), off }); |
| | 4449 | |
| | 4450 | self.updateSectionOrdinals(index + 1); |
| | 4451 | } |
| | 4452 | |
| | 4453 | return index; |
| | 4454 | } |
| | 4455 | |
| | 4456 | fn getSectionPrecedence(header: macho.section_64) u4 { |
| | 4457 | if (header.isCode()) { |
| | 4458 | if (mem.eql(u8, "__text", header.sectName())) return 0x0; |
| | 4459 | if (header.@"type"() == macho.S_SYMBOL_STUBS) return 0x1; |
| | 4460 | return 0x2; |
| | 4461 | } |
| | 4462 | switch (header.@"type"()) { |
| | 4463 | macho.S_NON_LAZY_SYMBOL_POINTERS, |
| | 4464 | macho.S_LAZY_SYMBOL_POINTERS, |
| | 4465 | => return 0x0, |
| | 4466 | macho.S_MOD_INIT_FUNC_POINTERS => return 0x1, |
| | 4467 | macho.S_MOD_TERM_FUNC_POINTERS => return 0x2, |
| | 4468 | macho.S_ZEROFILL => return 0xf, |
| | 4469 | macho.S_THREAD_LOCAL_REGULAR => return 0xd, |
| | 4470 | macho.S_THREAD_LOCAL_ZEROFILL => return 0xe, |
| | 4471 | else => if (mem.eql(u8, "__eh_frame", header.sectName())) |
| | 4472 | return 0xf |
| | 4473 | else |
| | 4474 | return 0x3, |
| 5007 | } | 4475 | } |
| | 4476 | } |
| 5008 | | 4477 | |
| 5009 | const index = @intCast(u16, seg.sections.items.len); | 4478 | fn insertSection(self: *MachO, segment_index: u8, header: macho.section_64) !u8 { |
| 5010 | try seg.sections.append(self.base.allocator, sect); | 4479 | const precedence = getSectionPrecedence(header); |
| 5011 | seg.inner.cmdsize += @sizeOf(macho.section_64); | 4480 | const indexes = self.getSectionIndexes(segment_index); |
| 5012 | seg.inner.nsects += 1; | 4481 | const insertion_index = for (self.sections.items(.header)[indexes.start..indexes.end]) |hdr, i| { |
| | 4482 | if (getSectionPrecedence(hdr) > precedence) break @intCast(u8, i + indexes.start); |
| | 4483 | } else indexes.end; |
| | 4484 | log.debug("inserting section '{s},{s}' at index {d}", .{ |
| | 4485 | header.segName(), |
| | 4486 | header.sectName(), |
| | 4487 | insertion_index, |
| | 4488 | }); |
| | 4489 | for (&[_]*?u8{ |
| | 4490 | &self.text_section_index, |
| | 4491 | &self.stubs_section_index, |
| | 4492 | &self.stub_helper_section_index, |
| | 4493 | &self.got_section_index, |
| | 4494 | &self.la_symbol_ptr_section_index, |
| | 4495 | &self.data_section_index, |
| | 4496 | }) |maybe_index| { |
| | 4497 | const index = maybe_index.* orelse continue; |
| | 4498 | if (insertion_index <= index) maybe_index.* = index + 1; |
| | 4499 | } |
| | 4500 | try self.sections.insert(self.base.allocator, insertion_index, .{ |
| | 4501 | .segment_index = segment_index, |
| | 4502 | .header = header, |
| | 4503 | }); |
| | 4504 | return insertion_index; |
| | 4505 | } |
| 5013 | | 4506 | |
| 5014 | const match = MatchingSection{ | 4507 | fn updateSectionOrdinals(self: *MachO, start: u8) void { |
| 5015 | .seg = segment_id, | 4508 | const tracy = trace(@src()); |
| 5016 | .sect = index, | 4509 | defer tracy.end(); |
| 5017 | }; | | |
| 5018 | _ = try self.section_ordinals.getOrPut(self.base.allocator, match); | | |
| 5019 | try self.atom_free_lists.putNoClobber(self.base.allocator, match, .{}); | | |
| 5020 | | 4510 | |
| 5021 | self.load_commands_dirty = true; | 4511 | const slice = self.sections.slice(); |
| 5022 | self.sections_order_dirty = true; | 4512 | for (slice.items(.last_atom)[start..]) |last_atom| { |
| | 4513 | var atom = last_atom orelse continue; |
| 5023 | | 4514 | |
| 5024 | return index; | 4515 | while (true) { |
| | 4516 | const sym = atom.getSymbolPtr(self); |
| | 4517 | sym.n_sect = start + 1; |
| | 4518 | |
| | 4519 | for (atom.contained.items) |sym_at_off| { |
| | 4520 | const contained_sym = self.getSymbolPtr(.{ |
| | 4521 | .sym_index = sym_at_off.sym_index, |
| | 4522 | .file = atom.file, |
| | 4523 | }); |
| | 4524 | contained_sym.n_sect = start + 1; |
| | 4525 | } |
| | 4526 | |
| | 4527 | if (atom.prev) |prev| { |
| | 4528 | atom = prev; |
| | 4529 | } else break; |
| | 4530 | } |
| | 4531 | } |
| 5025 | } | 4532 | } |
| 5026 | | 4533 | |
| 5027 | fn findFreeSpace(self: MachO, segment_id: u16, alignment: u64, start: ?u32) u64 { | 4534 | fn shiftLocalsByOffset(self: *MachO, sect_id: u8, offset: i64) !void { |
| 5028 | const seg = self.load_commands.items[segment_id].segment; | 4535 | var atom = self.sections.items(.last_atom)[sect_id] orelse return; |
| 5029 | if (seg.sections.items.len == 0) { | 4536 | |
| 5030 | return if (start) |v| v else seg.inner.fileoff; | 4537 | while (true) { |
| | 4538 | const atom_sym = atom.getSymbolPtr(self); |
| | 4539 | atom_sym.n_value = @intCast(u64, @intCast(i64, atom_sym.n_value) + offset); |
| | 4540 | |
| | 4541 | for (atom.contained.items) |sym_at_off| { |
| | 4542 | const contained_sym = self.getSymbolPtr(.{ |
| | 4543 | .sym_index = sym_at_off.sym_index, |
| | 4544 | .file = atom.file, |
| | 4545 | }); |
| | 4546 | contained_sym.n_value = @intCast(u64, @intCast(i64, contained_sym.n_value) + offset); |
| | 4547 | } |
| | 4548 | |
| | 4549 | if (atom.prev) |prev| { |
| | 4550 | atom = prev; |
| | 4551 | } else break; |
| 5031 | } | 4552 | } |
| 5032 | const last_sect = seg.sections.items[seg.sections.items.len - 1]; | | |
| 5033 | const final_off = last_sect.offset + padToIdeal(last_sect.size); | | |
| 5034 | return mem.alignForwardGeneric(u64, final_off, alignment); | | |
| 5035 | } | 4553 | } |
| 5036 | | 4554 | |
| 5037 | fn growSegment(self: *MachO, seg_id: u16, new_size: u64) !void { | 4555 | fn growSegment(self: *MachO, segment_index: u8, new_size: u64) !void { |
| 5038 | const seg = &self.load_commands.items[seg_id].segment; | 4556 | const segment = &self.segments.items[segment_index]; |
| 5039 | const new_seg_size = mem.alignForwardGeneric(u64, new_size, self.page_size); | 4557 | const new_segment_size = mem.alignForwardGeneric(u64, new_size, self.page_size); |
| 5040 | assert(new_seg_size > seg.inner.filesize); | 4558 | assert(new_segment_size > segment.filesize); |
| 5041 | const offset_amt = new_seg_size - seg.inner.filesize; | 4559 | const offset_amt = new_segment_size - segment.filesize; |
| 5042 | log.debug("growing segment {s} from 0x{x} to 0x{x}", .{ | 4560 | log.debug("growing segment {s} from 0x{x} to 0x{x}", .{ |
| 5043 | seg.inner.segname, | 4561 | segment.segname, |
| 5044 | seg.inner.filesize, | 4562 | segment.filesize, |
| 5045 | new_seg_size, | 4563 | new_segment_size, |
| 5046 | }); | 4564 | }); |
| 5047 | seg.inner.filesize = new_seg_size; | 4565 | segment.filesize = new_segment_size; |
| 5048 | seg.inner.vmsize = new_seg_size; | 4566 | segment.vmsize = new_segment_size; |
| 5049 | | 4567 | |
| 5050 | log.debug(" (new segment file offsets from 0x{x} to 0x{x} (in memory 0x{x} to 0x{x}))", .{ | 4568 | log.debug(" (new segment file offsets from 0x{x} to 0x{x} (in memory 0x{x} to 0x{x}))", .{ |
| 5051 | seg.inner.fileoff, | 4569 | segment.fileoff, |
| 5052 | seg.inner.fileoff + seg.inner.filesize, | 4570 | segment.fileoff + segment.filesize, |
| 5053 | seg.inner.vmaddr, | 4571 | segment.vmaddr, |
| 5054 | seg.inner.vmaddr + seg.inner.vmsize, | 4572 | segment.vmaddr + segment.vmsize, |
| 5055 | }); | 4573 | }); |
| 5056 | | 4574 | |
| 5057 | var next: usize = seg_id + 1; | 4575 | var next: u8 = segment_index + 1; |
| 5058 | while (next < self.linkedit_segment_cmd_index.? + 1) : (next += 1) { | 4576 | while (next < self.linkedit_segment_cmd_index.? + 1) : (next += 1) { |
| 5059 | const next_seg = &self.load_commands.items[next].segment; | 4577 | const next_segment = &self.segments.items[next]; |
| 5060 | | 4578 | |
| 5061 | try MachO.copyRangeAllOverlappingAlloc( | 4579 | try MachO.copyRangeAllOverlappingAlloc( |
| 5062 | self.base.allocator, | 4580 | self.base.allocator, |
| 5063 | self.base.file.?, | 4581 | self.base.file.?, |
| 5064 | next_seg.inner.fileoff, | 4582 | next_segment.fileoff, |
| 5065 | next_seg.inner.fileoff + offset_amt, | 4583 | next_segment.fileoff + offset_amt, |
| 5066 | math.cast(usize, next_seg.inner.filesize) orelse return error.Overflow, | 4584 | math.cast(usize, next_segment.filesize) orelse return error.Overflow, |
| 5067 | ); | 4585 | ); |
| 5068 | | 4586 | |
| 5069 | next_seg.inner.fileoff += offset_amt; | 4587 | next_segment.fileoff += offset_amt; |
| 5070 | next_seg.inner.vmaddr += offset_amt; | 4588 | next_segment.vmaddr += offset_amt; |
| 5071 | | 4589 | |
| 5072 | log.debug(" (new {s} segment file offsets from 0x{x} to 0x{x} (in memory 0x{x} to 0x{x}))", .{ | 4590 | log.debug(" (new {s} segment file offsets from 0x{x} to 0x{x} (in memory 0x{x} to 0x{x}))", .{ |
| 5073 | next_seg.inner.segname, | 4591 | next_segment.segname, |
| 5074 | next_seg.inner.fileoff, | 4592 | next_segment.fileoff, |
| 5075 | next_seg.inner.fileoff + next_seg.inner.filesize, | 4593 | next_segment.fileoff + next_segment.filesize, |
| 5076 | next_seg.inner.vmaddr, | 4594 | next_segment.vmaddr, |
| 5077 | next_seg.inner.vmaddr + next_seg.inner.vmsize, | 4595 | next_segment.vmaddr + next_segment.vmsize, |
| 5078 | }); | 4596 | }); |
| 5079 | | 4597 | |
| 5080 | for (next_seg.sections.items) |*moved_sect, moved_sect_id| { | 4598 | const indexes = self.getSectionIndexes(next); |
| 5081 | moved_sect.offset += @intCast(u32, offset_amt); | 4599 | for (self.sections.items(.header)[indexes.start..indexes.end]) |*header, i| { |
| 5082 | moved_sect.addr += offset_amt; | 4600 | header.offset += @intCast(u32, offset_amt); |
| 5083 | | 4601 | header.addr += offset_amt; |
| 5084 | log.debug(" (new {s},{s} file offsets from 0x{x} to 0x{x} (in memory 0x{x} to 0x{x}))", .{ | 4602 | |
| 5085 | moved_sect.segName(), | 4603 | log.debug(" (new {s},{s} file offsets from 0x{x} to 0x{x} (in memory 0x{x} to 0x{x}))", .{ |
| 5086 | moved_sect.sectName(), | 4604 | header.segName(), |
| 5087 | moved_sect.offset, | 4605 | header.sectName(), |
| 5088 | moved_sect.offset + moved_sect.size, | 4606 | header.offset, |
| 5089 | moved_sect.addr, | 4607 | header.offset + header.size, |
| 5090 | moved_sect.addr + moved_sect.size, | 4608 | header.addr, |
| | 4609 | header.addr + header.size, |
| 5091 | }); | 4610 | }); |
| 5092 | | 4611 | |
| 5093 | try self.shiftLocalsByOffset(.{ | 4612 | try self.shiftLocalsByOffset(@intCast(u8, i + indexes.start), @intCast(i64, offset_amt)); |
| 5094 | .seg = @intCast(u16, next), | | |
| 5095 | .sect = @intCast(u16, moved_sect_id), | | |
| 5096 | }, @intCast(i64, offset_amt)); | | |
| 5097 | } | 4613 | } |
| 5098 | } | 4614 | } |
| 5099 | } | 4615 | } |
| 5100 | | 4616 | |
| 5101 | fn growSection(self: *MachO, match: MatchingSection, new_size: u32) !void { | 4617 | fn growSection(self: *MachO, sect_id: u8, new_size: u32) !void { |
| 5102 | const tracy = trace(@src()); | 4618 | const tracy = trace(@src()); |
| 5103 | defer tracy.end(); | 4619 | defer tracy.end(); |
| 5104 | | 4620 | |
| 5105 | const seg = &self.load_commands.items[match.seg].segment; | 4621 | const section = self.sections.get(sect_id); |
| 5106 | const sect = &seg.sections.items[match.sect]; | 4622 | const segment_index = section.segment_index; |
| | 4623 | const header = section.header; |
| | 4624 | const segment = self.segments.items[segment_index]; |
| 5107 | | 4625 | |
| 5108 | const alignment = try math.powi(u32, 2, sect.@"align"); | 4626 | const alignment = try math.powi(u32, 2, header.@"align"); |
| 5109 | const max_size = self.allocatedSize(match.seg, sect.offset); | 4627 | const max_size = self.allocatedSize(segment_index, header.offset); |
| 5110 | const ideal_size = padToIdeal(new_size); | 4628 | const ideal_size = padToIdeal(new_size); |
| 5111 | const needed_size = mem.alignForwardGeneric(u32, ideal_size, alignment); | 4629 | const needed_size = mem.alignForwardGeneric(u32, ideal_size, alignment); |
| 5112 | | 4630 | |
| 5113 | if (needed_size > max_size) blk: { | 4631 | if (needed_size > max_size) blk: { |
| 5114 | log.debug(" (need to grow! needed 0x{x}, max 0x{x})", .{ needed_size, max_size }); | 4632 | log.debug(" (need to grow! needed 0x{x}, max 0x{x})", .{ needed_size, max_size }); |
| 5115 | | 4633 | |
| 5116 | if (match.sect == seg.sections.items.len - 1) { | 4634 | const indexes = self.getSectionIndexes(segment_index); |
| | 4635 | if (sect_id == indexes.end - 1) { |
| 5117 | // Last section, just grow segments | 4636 | // Last section, just grow segments |
| 5118 | try self.growSegment(match.seg, seg.inner.filesize + needed_size - max_size); | 4637 | try self.growSegment(segment_index, segment.filesize + needed_size - max_size); |
| 5119 | break :blk; | 4638 | break :blk; |
| 5120 | } | 4639 | } |
| 5121 | | 4640 | |
| 5122 | // Need to move all sections below in file and address spaces. | 4641 | // Need to move all sections below in file and address spaces. |
| 5123 | const offset_amt = offset: { | 4642 | const offset_amt = offset: { |
| 5124 | const max_alignment = try self.getSectionMaxAlignment(match.seg, match.sect + 1); | 4643 | const max_alignment = try self.getSectionMaxAlignment(sect_id + 1, indexes.end); |
| 5125 | break :offset mem.alignForwardGeneric(u64, needed_size - max_size, max_alignment); | 4644 | break :offset mem.alignForwardGeneric(u64, needed_size - max_size, max_alignment); |
| 5126 | }; | 4645 | }; |
| 5127 | | 4646 | |
| 5128 | // Before we commit to this, check if the segment needs to grow too. | 4647 | // Before we commit to this, check if the segment needs to grow too. |
| 5129 | // We assume that each section header is growing linearly with the increasing | 4648 | // We assume that each section header is growing linearly with the increasing |
| 5130 | // file offset / virtual memory address space. | 4649 | // file offset / virtual memory address space. |
| 5131 | const last_sect = seg.sections.items[seg.sections.items.len - 1]; | 4650 | const last_sect_header = self.sections.items(.header)[indexes.end - 1]; |
| 5132 | const last_sect_off = last_sect.offset + last_sect.size; | 4651 | const last_sect_off = last_sect_header.offset + last_sect_header.size; |
| 5133 | const seg_off = seg.inner.fileoff + seg.inner.filesize; | 4652 | const seg_off = segment.fileoff + segment.filesize; |
| 5134 | | 4653 | |
| 5135 | if (last_sect_off + offset_amt > seg_off) { | 4654 | if (last_sect_off + offset_amt > seg_off) { |
| 5136 | // Need to grow segment first. | 4655 | // Need to grow segment first. |
| 5137 | const spill_size = (last_sect_off + offset_amt) - seg_off; | 4656 | const spill_size = (last_sect_off + offset_amt) - seg_off; |
| 5138 | try self.growSegment(match.seg, seg.inner.filesize + spill_size); | 4657 | try self.growSegment(segment_index, segment.filesize + spill_size); |
| 5139 | } | 4658 | } |
| 5140 | | 4659 | |
| 5141 | // We have enough space to expand within the segment, so move all sections by | 4660 | // We have enough space to expand within the segment, so move all sections by |
| 5142 | // the required amount and update their header offsets. | 4661 | // the required amount and update their header offsets. |
| 5143 | const next_sect = seg.sections.items[match.sect + 1]; | 4662 | const next_sect = self.sections.items(.header)[sect_id + 1]; |
| 5144 | const total_size = last_sect_off - next_sect.offset; | 4663 | const total_size = last_sect_off - next_sect.offset; |
| 5145 | | 4664 | |
| 5146 | try MachO.copyRangeAllOverlappingAlloc( | 4665 | try MachO.copyRangeAllOverlappingAlloc( |
| ... | @@ -5151,9 +4670,7 @@ fn growSection(self: *MachO, match: MatchingSection, new_size: u32) !void { | ... | @@ -5151,9 +4670,7 @@ fn growSection(self: *MachO, match: MatchingSection, new_size: u32) !void { |
| 5151 | math.cast(usize, total_size) orelse return error.Overflow, | 4670 | math.cast(usize, total_size) orelse return error.Overflow, |
| 5152 | ); | 4671 | ); |
| 5153 | | 4672 | |
| 5154 | var next = match.sect + 1; | 4673 | for (self.sections.items(.header)[sect_id + 1 .. indexes.end]) |*moved_sect, i| { |
| 5155 | while (next < seg.sections.items.len) : (next += 1) { | | |
| 5156 | const moved_sect = &seg.sections.items[next]; | | |
| 5157 | moved_sect.offset += @intCast(u32, offset_amt); | 4674 | moved_sect.offset += @intCast(u32, offset_amt); |
| 5158 | moved_sect.addr += offset_amt; | 4675 | moved_sect.addr += offset_amt; |
| 5159 | | 4676 | |
| ... | @@ -5166,49 +4683,45 @@ fn growSection(self: *MachO, match: MatchingSection, new_size: u32) !void { | ... | @@ -5166,49 +4683,45 @@ fn growSection(self: *MachO, match: MatchingSection, new_size: u32) !void { |
| 5166 | moved_sect.addr + moved_sect.size, | 4683 | moved_sect.addr + moved_sect.size, |
| 5167 | }); | 4684 | }); |
| 5168 | | 4685 | |
| 5169 | try self.shiftLocalsByOffset(.{ | 4686 | try self.shiftLocalsByOffset(@intCast(u8, sect_id + 1 + i), @intCast(i64, offset_amt)); |
| 5170 | .seg = match.seg, | | |
| 5171 | .sect = next, | | |
| 5172 | }, @intCast(i64, offset_amt)); | | |
| 5173 | } | 4687 | } |
| 5174 | } | 4688 | } |
| 5175 | } | 4689 | } |
| 5176 | | 4690 | |
| 5177 | fn allocatedSize(self: MachO, segment_id: u16, start: u64) u64 { | 4691 | fn allocatedSize(self: MachO, segment_id: u8, start: u64) u64 { |
| 5178 | const seg = self.load_commands.items[segment_id].segment; | 4692 | const segment = self.segments.items[segment_id]; |
| 5179 | assert(start >= seg.inner.fileoff); | 4693 | const indexes = self.getSectionIndexes(segment_id); |
| 5180 | var min_pos: u64 = seg.inner.fileoff + seg.inner.filesize; | 4694 | assert(start >= segment.fileoff); |
| | 4695 | var min_pos: u64 = segment.fileoff + segment.filesize; |
| 5181 | if (start > min_pos) return 0; | 4696 | if (start > min_pos) return 0; |
| 5182 | for (seg.sections.items) |section| { | 4697 | for (self.sections.items(.header)[indexes.start..indexes.end]) |header| { |
| 5183 | if (section.offset <= start) continue; | 4698 | if (header.offset <= start) continue; |
| 5184 | if (section.offset < min_pos) min_pos = section.offset; | 4699 | if (header.offset < min_pos) min_pos = header.offset; |
| 5185 | } | 4700 | } |
| 5186 | return min_pos - start; | 4701 | return min_pos - start; |
| 5187 | } | 4702 | } |
| 5188 | | 4703 | |
| 5189 | fn getSectionMaxAlignment(self: *MachO, segment_id: u16, start_sect_id: u16) !u32 { | 4704 | fn getSectionMaxAlignment(self: *MachO, start: u8, end: u8) !u32 { |
| 5190 | const seg = self.load_commands.items[segment_id].segment; | | |
| 5191 | var max_alignment: u32 = 1; | 4705 | var max_alignment: u32 = 1; |
| 5192 | var next = start_sect_id; | 4706 | const slice = self.sections.slice(); |
| 5193 | while (next < seg.sections.items.len) : (next += 1) { | 4707 | for (slice.items(.header)[start..end]) |header| { |
| 5194 | const sect = seg.sections.items[next]; | 4708 | const alignment = try math.powi(u32, 2, header.@"align"); |
| 5195 | const alignment = try math.powi(u32, 2, sect.@"align"); | | |
| 5196 | max_alignment = math.max(max_alignment, alignment); | 4709 | max_alignment = math.max(max_alignment, alignment); |
| 5197 | } | 4710 | } |
| 5198 | return max_alignment; | 4711 | return max_alignment; |
| 5199 | } | 4712 | } |
| 5200 | | 4713 | |
| 5201 | fn allocateAtomCommon(self: *MachO, atom: *Atom, match: MatchingSection) !void { | 4714 | fn allocateAtomCommon(self: *MachO, atom: *Atom, sect_id: u8) !void { |
| 5202 | const sym = atom.getSymbolPtr(self); | 4715 | const sym = atom.getSymbolPtr(self); |
| 5203 | if (self.mode == .incremental) { | 4716 | if (self.mode == .incremental) { |
| 5204 | const size = atom.size; | 4717 | const size = atom.size; |
| 5205 | const alignment = try math.powi(u32, 2, atom.alignment); | 4718 | const alignment = try math.powi(u32, 2, atom.alignment); |
| 5206 | const vaddr = try self.allocateAtom(atom, size, alignment, match); | 4719 | const vaddr = try self.allocateAtom(atom, size, alignment, sect_id); |
| 5207 | const sym_name = atom.getName(self); | 4720 | const sym_name = atom.getName(self); |
| 5208 | log.debug("allocated {s} atom at 0x{x}", .{ sym_name, vaddr }); | 4721 | log.debug("allocated {s} atom at 0x{x}", .{ sym_name, vaddr }); |
| 5209 | sym.n_value = vaddr; | 4722 | sym.n_value = vaddr; |
| 5210 | } else try self.addAtomToSection(atom, match); | 4723 | } else try self.addAtomToSection(atom, sect_id); |
| 5211 | sym.n_sect = self.getSectionOrdinal(match); | 4724 | sym.n_sect = sect_id + 1; |
| 5212 | } | 4725 | } |
| 5213 | | 4726 | |
| 5214 | fn allocateAtom( | 4727 | fn allocateAtom( |
| ... | @@ -5216,15 +4729,15 @@ fn allocateAtom( | ... | @@ -5216,15 +4729,15 @@ fn allocateAtom( |
| 5216 | atom: *Atom, | 4729 | atom: *Atom, |
| 5217 | new_atom_size: u64, | 4730 | new_atom_size: u64, |
| 5218 | alignment: u64, | 4731 | alignment: u64, |
| 5219 | match: MatchingSection, | 4732 | sect_id: u8, |
| 5220 | ) !u64 { | 4733 | ) !u64 { |
| 5221 | const tracy = trace(@src()); | 4734 | const tracy = trace(@src()); |
| 5222 | defer tracy.end(); | 4735 | defer tracy.end(); |
| 5223 | | 4736 | |
| 5224 | const sect = self.getSectionPtr(match); | 4737 | const header = &self.sections.items(.header)[sect_id]; |
| 5225 | var free_list = self.atom_free_lists.get(match).?; | 4738 | const free_list = &self.sections.items(.free_list)[sect_id]; |
| 5226 | const needs_padding = match.seg == self.text_segment_cmd_index.? and match.sect == self.text_section_index.?; | 4739 | const maybe_last_atom = &self.sections.items(.last_atom)[sect_id]; |
| 5227 | const new_atom_ideal_capacity = if (needs_padding) padToIdeal(new_atom_size) else new_atom_size; | 4740 | const new_atom_ideal_capacity = if (header.isCode()) padToIdeal(new_atom_size) else new_atom_size; |
| 5228 | | 4741 | |
| 5229 | // We use these to indicate our intention to update metadata, placing the new atom, | 4742 | // We use these to indicate our intention to update metadata, placing the new atom, |
| 5230 | // and possibly removing a free list node. | 4743 | // and possibly removing a free list node. |
| ... | @@ -5244,7 +4757,7 @@ fn allocateAtom( | ... | @@ -5244,7 +4757,7 @@ fn allocateAtom( |
| 5244 | // Is it enough that we could fit this new atom? | 4757 | // Is it enough that we could fit this new atom? |
| 5245 | const sym = big_atom.getSymbol(self); | 4758 | const sym = big_atom.getSymbol(self); |
| 5246 | const capacity = big_atom.capacity(self); | 4759 | const capacity = big_atom.capacity(self); |
| 5247 | const ideal_capacity = if (needs_padding) padToIdeal(capacity) else capacity; | 4760 | const ideal_capacity = if (header.isCode()) padToIdeal(capacity) else capacity; |
| 5248 | const ideal_capacity_end_vaddr = math.add(u64, sym.n_value, ideal_capacity) catch ideal_capacity; | 4761 | const ideal_capacity_end_vaddr = math.add(u64, sym.n_value, ideal_capacity) catch ideal_capacity; |
| 5249 | const capacity_end_vaddr = sym.n_value + capacity; | 4762 | const capacity_end_vaddr = sym.n_value + capacity; |
| 5250 | const new_start_vaddr_unaligned = capacity_end_vaddr - new_atom_ideal_capacity; | 4763 | const new_start_vaddr_unaligned = capacity_end_vaddr - new_atom_ideal_capacity; |
| ... | @@ -5272,30 +4785,28 @@ fn allocateAtom( | ... | @@ -5272,30 +4785,28 @@ fn allocateAtom( |
| 5272 | free_list_removal = i; | 4785 | free_list_removal = i; |
| 5273 | } | 4786 | } |
| 5274 | break :blk new_start_vaddr; | 4787 | break :blk new_start_vaddr; |
| 5275 | } else if (self.atoms.get(match)) |last| { | 4788 | } else if (maybe_last_atom.*) |last| { |
| 5276 | const last_symbol = last.getSymbol(self); | 4789 | const last_symbol = last.getSymbol(self); |
| 5277 | const ideal_capacity = if (needs_padding) padToIdeal(last.size) else last.size; | 4790 | const ideal_capacity = if (header.isCode()) padToIdeal(last.size) else last.size; |
| 5278 | const ideal_capacity_end_vaddr = last_symbol.n_value + ideal_capacity; | 4791 | const ideal_capacity_end_vaddr = last_symbol.n_value + ideal_capacity; |
| 5279 | const new_start_vaddr = mem.alignForwardGeneric(u64, ideal_capacity_end_vaddr, alignment); | 4792 | const new_start_vaddr = mem.alignForwardGeneric(u64, ideal_capacity_end_vaddr, alignment); |
| 5280 | atom_placement = last; | 4793 | atom_placement = last; |
| 5281 | break :blk new_start_vaddr; | 4794 | break :blk new_start_vaddr; |
| 5282 | } else { | 4795 | } else { |
| 5283 | break :blk mem.alignForwardGeneric(u64, sect.addr, alignment); | 4796 | break :blk mem.alignForwardGeneric(u64, header.addr, alignment); |
| 5284 | } | 4797 | } |
| 5285 | }; | 4798 | }; |
| 5286 | | 4799 | |
| 5287 | const expand_section = atom_placement == null or atom_placement.?.next == null; | 4800 | const expand_section = atom_placement == null or atom_placement.?.next == null; |
| 5288 | if (expand_section) { | 4801 | if (expand_section) { |
| 5289 | const needed_size = @intCast(u32, (vaddr + new_atom_size) - sect.addr); | 4802 | const needed_size = @intCast(u32, (vaddr + new_atom_size) - header.addr); |
| 5290 | try self.growSection(match, needed_size); | 4803 | try self.growSection(sect_id, needed_size); |
| 5291 | _ = try self.atoms.put(self.base.allocator, match, atom); | 4804 | maybe_last_atom.* = atom; |
| 5292 | sect.size = needed_size; | 4805 | header.size = needed_size; |
| 5293 | self.load_commands_dirty = true; | | |
| 5294 | } | 4806 | } |
| 5295 | const align_pow = @intCast(u32, math.log2(alignment)); | 4807 | const align_pow = @intCast(u32, math.log2(alignment)); |
| 5296 | if (sect.@"align" < align_pow) { | 4808 | if (header.@"align" < align_pow) { |
| 5297 | sect.@"align" = align_pow; | 4809 | header.@"align" = align_pow; |
| 5298 | self.load_commands_dirty = true; | | |
| 5299 | } | 4810 | } |
| 5300 | atom.size = new_atom_size; | 4811 | atom.size = new_atom_size; |
| 5301 | atom.alignment = align_pow; | 4812 | atom.alignment = align_pow; |
| ... | @@ -5322,20 +4833,19 @@ fn allocateAtom( | ... | @@ -5322,20 +4833,19 @@ fn allocateAtom( |
| 5322 | return vaddr; | 4833 | return vaddr; |
| 5323 | } | 4834 | } |
| 5324 | | 4835 | |
| 5325 | pub fn addAtomToSection(self: *MachO, atom: *Atom, match: MatchingSection) !void { | 4836 | pub fn addAtomToSection(self: *MachO, atom: *Atom, sect_id: u8) !void { |
| 5326 | if (self.atoms.getPtr(match)) |last| { | 4837 | var section = self.sections.get(sect_id); |
| 5327 | last.*.next = atom; | 4838 | if (section.header.size > 0) { |
| 5328 | atom.prev = last.*; | 4839 | section.last_atom.?.next = atom; |
| 5329 | last.* = atom; | 4840 | atom.prev = section.last_atom.?; |
| 5330 | } else { | | |
| 5331 | try self.atoms.putNoClobber(self.base.allocator, match, atom); | | |
| 5332 | } | 4841 | } |
| 5333 | const sect = self.getSectionPtr(match); | 4842 | section.last_atom = atom; |
| 5334 | const atom_alignment = try math.powi(u32, 2, atom.alignment); | 4843 | const atom_alignment = try math.powi(u32, 2, atom.alignment); |
| 5335 | const aligned_end_addr = mem.alignForwardGeneric(u64, sect.size, atom_alignment); | 4844 | const aligned_end_addr = mem.alignForwardGeneric(u64, section.header.size, atom_alignment); |
| 5336 | const padding = aligned_end_addr - sect.size; | 4845 | const padding = aligned_end_addr - section.header.size; |
| 5337 | sect.size += padding + atom.size; | 4846 | section.header.size += padding + atom.size; |
| 5338 | sect.@"align" = @maximum(sect.@"align", atom.alignment); | 4847 | section.header.@"align" = @maximum(section.header.@"align", atom.alignment); |
| | 4848 | self.sections.set(sect_id, section); |
| 5339 | } | 4849 | } |
| 5340 | | 4850 | |
| 5341 | pub fn getGlobalSymbol(self: *MachO, name: []const u8) !u32 { | 4851 | pub fn getGlobalSymbol(self: *MachO, name: []const u8) !u32 { |
| ... | @@ -5368,208 +4878,61 @@ pub fn getGlobalSymbol(self: *MachO, name: []const u8) !u32 { | ... | @@ -5368,208 +4878,61 @@ pub fn getGlobalSymbol(self: *MachO, name: []const u8) !u32 { |
| 5368 | return sym_index; | 4878 | return sym_index; |
| 5369 | } | 4879 | } |
| 5370 | | 4880 | |
| 5371 | fn getSegmentAllocBase(self: MachO, indices: []const ?u16) struct { vmaddr: u64, fileoff: u64 } { | 4881 | fn getSegmentAllocBase(self: MachO, indices: []const ?u8) struct { vmaddr: u64, fileoff: u64 } { |
| 5372 | for (indices) |maybe_prev_id| { | 4882 | for (indices) |maybe_prev_id| { |
| 5373 | const prev_id = maybe_prev_id orelse continue; | 4883 | const prev_id = maybe_prev_id orelse continue; |
| 5374 | const prev = self.load_commands.items[prev_id].segment; | 4884 | const prev = self.segments.items[prev_id]; |
| 5375 | return .{ | 4885 | return .{ |
| 5376 | .vmaddr = prev.inner.vmaddr + prev.inner.vmsize, | 4886 | .vmaddr = prev.vmaddr + prev.vmsize, |
| 5377 | .fileoff = prev.inner.fileoff + prev.inner.filesize, | 4887 | .fileoff = prev.fileoff + prev.filesize, |
| 5378 | }; | 4888 | }; |
| 5379 | } | 4889 | } |
| 5380 | return .{ .vmaddr = 0, .fileoff = 0 }; | 4890 | return .{ .vmaddr = 0, .fileoff = 0 }; |
| 5381 | } | 4891 | } |
| 5382 | | 4892 | |
| 5383 | fn pruneAndSortSectionsInSegment(self: *MachO, maybe_seg_id: *?u16, indices: []*?u16) !void { | 4893 | fn writeSegmentHeaders(self: *MachO, ncmds: *u32, writer: anytype) !void { |
| 5384 | const seg_id = maybe_seg_id.* orelse return; | 4894 | for (self.segments.items) |seg, i| { |
| 5385 | | 4895 | const indexes = self.getSectionIndexes(@intCast(u8, i)); |
| 5386 | var mapping = std.AutoArrayHashMap(u16, ?u16).init(self.base.allocator); | 4896 | var out_seg = seg; |
| 5387 | defer mapping.deinit(); | 4897 | out_seg.cmdsize = @sizeOf(macho.segment_command_64); |
| 5388 | | 4898 | out_seg.nsects = 0; |
| 5389 | const seg = &self.load_commands.items[seg_id].segment; | 4899 | |
| 5390 | var sections = seg.sections.toOwnedSlice(self.base.allocator); | 4900 | // Update section headers count; any section with size of 0 is excluded |
| 5391 | defer self.base.allocator.free(sections); | 4901 | // since it doesn't have any data in the final binary file. |
| 5392 | try seg.sections.ensureTotalCapacity(self.base.allocator, sections.len); | 4902 | for (self.sections.items(.header)[indexes.start..indexes.end]) |header| { |
| 5393 | | 4903 | if (header.size == 0) continue; |
| 5394 | for (indices) |maybe_index| { | 4904 | out_seg.cmdsize += @sizeOf(macho.section_64); |
| 5395 | const old_idx = maybe_index.* orelse continue; | 4905 | out_seg.nsects += 1; |
| 5396 | const sect = &sections[old_idx]; | | |
| 5397 | if (sect.size == 0) { | | |
| 5398 | log.debug("pruning section {s},{s}", .{ sect.segName(), sect.sectName() }); | | |
| 5399 | maybe_index.* = null; | | |
| 5400 | seg.inner.cmdsize -= @sizeOf(macho.section_64); | | |
| 5401 | seg.inner.nsects -= 1; | | |
| 5402 | } else { | | |
| 5403 | maybe_index.* = @intCast(u16, seg.sections.items.len); | | |
| 5404 | seg.sections.appendAssumeCapacity(sect.*); | | |
| 5405 | } | 4906 | } |
| 5406 | try mapping.putNoClobber(old_idx, maybe_index.*); | | |
| 5407 | } | | |
| 5408 | | | |
| 5409 | var atoms = std.ArrayList(struct { match: MatchingSection, atom: *Atom }).init(self.base.allocator); | | |
| 5410 | defer atoms.deinit(); | | |
| 5411 | try atoms.ensureTotalCapacity(mapping.count()); | | |
| 5412 | | | |
| 5413 | for (mapping.keys()) |old_sect| { | | |
| 5414 | const new_sect = mapping.get(old_sect).? orelse { | | |
| 5415 | _ = self.atoms.remove(.{ .seg = seg_id, .sect = old_sect }); | | |
| 5416 | continue; | | |
| 5417 | }; | | |
| 5418 | const kv = self.atoms.fetchRemove(.{ .seg = seg_id, .sect = old_sect }).?; | | |
| 5419 | atoms.appendAssumeCapacity(.{ | | |
| 5420 | .match = .{ .seg = seg_id, .sect = new_sect }, | | |
| 5421 | .atom = kv.value, | | |
| 5422 | }); | | |
| 5423 | } | | |
| 5424 | | | |
| 5425 | while (atoms.popOrNull()) |next| { | | |
| 5426 | try self.atoms.putNoClobber(self.base.allocator, next.match, next.atom); | | |
| 5427 | } | | |
| 5428 | | | |
| 5429 | if (seg.inner.nsects == 0 and !mem.eql(u8, "__TEXT", seg.inner.segName())) { | | |
| 5430 | // Segment has now become empty, so mark it as such | | |
| 5431 | log.debug("marking segment {s} as dead", .{seg.inner.segName()}); | | |
| 5432 | seg.inner.cmd = @intToEnum(macho.LC, 0); | | |
| 5433 | maybe_seg_id.* = null; | | |
| 5434 | } | | |
| 5435 | } | | |
| 5436 | | 4907 | |
| 5437 | fn pruneAndSortSections(self: *MachO) !void { | 4908 | if (out_seg.nsects == 0 and |
| 5438 | try self.pruneAndSortSectionsInSegment(&self.text_segment_cmd_index, &.{ | 4909 | (mem.eql(u8, out_seg.segName(), "__DATA_CONST") or |
| 5439 | &self.text_section_index, | 4910 | mem.eql(u8, out_seg.segName(), "__DATA"))) continue; |
| 5440 | &self.stubs_section_index, | | |
| 5441 | &self.stub_helper_section_index, | | |
| 5442 | &self.gcc_except_tab_section_index, | | |
| 5443 | &self.cstring_section_index, | | |
| 5444 | &self.ustring_section_index, | | |
| 5445 | &self.text_const_section_index, | | |
| 5446 | &self.objc_methlist_section_index, | | |
| 5447 | &self.objc_methname_section_index, | | |
| 5448 | &self.objc_methtype_section_index, | | |
| 5449 | &self.objc_classname_section_index, | | |
| 5450 | &self.eh_frame_section_index, | | |
| 5451 | }); | | |
| 5452 | | | |
| 5453 | try self.pruneAndSortSectionsInSegment(&self.data_const_segment_cmd_index, &.{ | | |
| 5454 | &self.got_section_index, | | |
| 5455 | &self.mod_init_func_section_index, | | |
| 5456 | &self.mod_term_func_section_index, | | |
| 5457 | &self.data_const_section_index, | | |
| 5458 | &self.objc_cfstring_section_index, | | |
| 5459 | &self.objc_classlist_section_index, | | |
| 5460 | &self.objc_imageinfo_section_index, | | |
| 5461 | }); | | |
| 5462 | | | |
| 5463 | try self.pruneAndSortSectionsInSegment(&self.data_segment_cmd_index, &.{ | | |
| 5464 | &self.rustc_section_index, | | |
| 5465 | &self.la_symbol_ptr_section_index, | | |
| 5466 | &self.objc_const_section_index, | | |
| 5467 | &self.objc_selrefs_section_index, | | |
| 5468 | &self.objc_classrefs_section_index, | | |
| 5469 | &self.objc_data_section_index, | | |
| 5470 | &self.data_section_index, | | |
| 5471 | &self.tlv_section_index, | | |
| 5472 | &self.tlv_ptrs_section_index, | | |
| 5473 | &self.tlv_data_section_index, | | |
| 5474 | &self.tlv_bss_section_index, | | |
| 5475 | &self.bss_section_index, | | |
| 5476 | }); | | |
| 5477 | | 4911 | |
| 5478 | // Create new section ordinals. | 4912 | try writer.writeStruct(out_seg); |
| 5479 | self.section_ordinals.clearRetainingCapacity(); | 4913 | for (self.sections.items(.header)[indexes.start..indexes.end]) |header| { |
| 5480 | if (self.text_segment_cmd_index) |seg_id| { | 4914 | if (header.size == 0) continue; |
| 5481 | const seg = self.load_commands.items[seg_id].segment; | 4915 | try writer.writeStruct(header); |
| 5482 | for (seg.sections.items) |_, sect_id| { | | |
| 5483 | const res = self.section_ordinals.getOrPutAssumeCapacity(.{ | | |
| 5484 | .seg = seg_id, | | |
| 5485 | .sect = @intCast(u16, sect_id), | | |
| 5486 | }); | | |
| 5487 | assert(!res.found_existing); | | |
| 5488 | } | | |
| 5489 | } | | |
| 5490 | if (self.data_const_segment_cmd_index) |seg_id| { | | |
| 5491 | const seg = self.load_commands.items[seg_id].segment; | | |
| 5492 | for (seg.sections.items) |_, sect_id| { | | |
| 5493 | const res = self.section_ordinals.getOrPutAssumeCapacity(.{ | | |
| 5494 | .seg = seg_id, | | |
| 5495 | .sect = @intCast(u16, sect_id), | | |
| 5496 | }); | | |
| 5497 | assert(!res.found_existing); | | |
| 5498 | } | | |
| 5499 | } | | |
| 5500 | if (self.data_segment_cmd_index) |seg_id| { | | |
| 5501 | const seg = self.load_commands.items[seg_id].segment; | | |
| 5502 | for (seg.sections.items) |_, sect_id| { | | |
| 5503 | const res = self.section_ordinals.getOrPutAssumeCapacity(.{ | | |
| 5504 | .seg = seg_id, | | |
| 5505 | .sect = @intCast(u16, sect_id), | | |
| 5506 | }); | | |
| 5507 | assert(!res.found_existing); | | |
| 5508 | } | 4916 | } |
| | 4917 | |
| | 4918 | ncmds.* += 1; |
| 5509 | } | 4919 | } |
| 5510 | self.sections_order_dirty = false; | | |
| 5511 | } | 4920 | } |
| 5512 | | 4921 | |
| 5513 | fn updateSectionOrdinals(self: *MachO) !void { | 4922 | fn writeLinkeditSegmentData(self: *MachO, ncmds: *u32, lc_writer: anytype) !void { |
| 5514 | if (!self.sections_order_dirty) return; | 4923 | const seg = &self.segments.items[self.linkedit_segment_cmd_index.?]; |
| 5515 | | 4924 | seg.filesize = 0; |
| 5516 | const tracy = trace(@src()); | 4925 | seg.vmsize = 0; |
| 5517 | defer tracy.end(); | | |
| 5518 | | | |
| 5519 | log.debug("updating section ordinals", .{}); | | |
| 5520 | | | |
| 5521 | const gpa = self.base.allocator; | | |
| 5522 | | | |
| 5523 | var ordinal_remap = std.AutoHashMap(u8, u8).init(gpa); | | |
| 5524 | defer ordinal_remap.deinit(); | | |
| 5525 | var ordinals: std.AutoArrayHashMapUnmanaged(MatchingSection, void) = .{}; | | |
| 5526 | | | |
| 5527 | var new_ordinal: u8 = 0; | | |
| 5528 | for (&[_]?u16{ | | |
| 5529 | self.text_segment_cmd_index, | | |
| 5530 | self.data_const_segment_cmd_index, | | |
| 5531 | self.data_segment_cmd_index, | | |
| 5532 | }) |maybe_index| { | | |
| 5533 | const index = maybe_index orelse continue; | | |
| 5534 | const seg = self.load_commands.items[index].segment; | | |
| 5535 | for (seg.sections.items) |sect, sect_id| { | | |
| 5536 | const match = MatchingSection{ | | |
| 5537 | .seg = @intCast(u16, index), | | |
| 5538 | .sect = @intCast(u16, sect_id), | | |
| 5539 | }; | | |
| 5540 | const old_ordinal = self.getSectionOrdinal(match); | | |
| 5541 | new_ordinal += 1; | | |
| 5542 | log.debug("'{s},{s}': sect({d}, '_,_') => sect({d}, '_,_')", .{ | | |
| 5543 | sect.segName(), | | |
| 5544 | sect.sectName(), | | |
| 5545 | old_ordinal, | | |
| 5546 | new_ordinal, | | |
| 5547 | }); | | |
| 5548 | try ordinal_remap.putNoClobber(old_ordinal, new_ordinal); | | |
| 5549 | try ordinals.putNoClobber(gpa, match, {}); | | |
| 5550 | } | | |
| 5551 | } | | |
| 5552 | | 4926 | |
| 5553 | // FIXME Jakub | 4927 | try self.writeDyldInfoData(ncmds, lc_writer); |
| 5554 | // TODO no need for duping work here; simply walk the atom graph | 4928 | try self.writeFunctionStarts(ncmds, lc_writer); |
| 5555 | for (self.locals.items) |*sym| { | 4929 | try self.writeDataInCode(ncmds, lc_writer); |
| 5556 | if (sym.undf()) continue; | 4930 | try self.writeSymtabs(ncmds, lc_writer); |
| 5557 | if (sym.n_sect == 0) continue; | | |
| 5558 | sym.n_sect = ordinal_remap.get(sym.n_sect).?; | | |
| 5559 | } | | |
| 5560 | for (self.objects.items) |*object| { | | |
| 5561 | for (object.symtab.items) |*sym| { | | |
| 5562 | if (sym.undf()) continue; | | |
| 5563 | if (sym.n_sect == 0) continue; | | |
| 5564 | sym.n_sect = ordinal_remap.get(sym.n_sect).?; | | |
| 5565 | } | | |
| 5566 | } | | |
| 5567 | | 4931 | |
| 5568 | self.section_ordinals.deinit(gpa); | 4932 | seg.vmsize = mem.alignForwardGeneric(u64, seg.filesize, self.page_size); |
| 5569 | self.section_ordinals = ordinals; | | |
| 5570 | } | 4933 | } |
| 5571 | | 4934 | |
| 5572 | fn writeDyldInfoData(self: *MachO) !void { | 4935 | fn writeDyldInfoData(self: *MachO, ncmds: *u32, lc_writer: anytype) !void { |
| 5573 | const tracy = trace(@src()); | 4936 | const tracy = trace(@src()); |
| 5574 | defer tracy.end(); | 4937 | defer tracy.end(); |
| 5575 | | 4938 | |
| ... | @@ -5582,89 +4945,86 @@ fn writeDyldInfoData(self: *MachO) !void { | ... | @@ -5582,89 +4945,86 @@ fn writeDyldInfoData(self: *MachO) !void { |
| 5582 | var lazy_bind_pointers = std.ArrayList(bind.Pointer).init(gpa); | 4945 | var lazy_bind_pointers = std.ArrayList(bind.Pointer).init(gpa); |
| 5583 | defer lazy_bind_pointers.deinit(); | 4946 | defer lazy_bind_pointers.deinit(); |
| 5584 | | 4947 | |
| 5585 | { | 4948 | const slice = self.sections.slice(); |
| 5586 | var it = self.atoms.iterator(); | 4949 | for (slice.items(.last_atom)) |last_atom, sect_id| { |
| 5587 | while (it.next()) |entry| { | 4950 | var atom = last_atom orelse continue; |
| 5588 | const match = entry.key_ptr.*; | 4951 | const segment_index = slice.items(.segment_index)[sect_id]; |
| 5589 | var atom: *Atom = entry.value_ptr.*; | 4952 | const header = slice.items(.header)[sect_id]; |
| 5590 | | 4953 | |
| 5591 | if (self.text_segment_cmd_index) |seg| { | 4954 | if (mem.eql(u8, header.segName(), "__TEXT")) continue; // __TEXT is non-writable |
| 5592 | if (match.seg == seg) continue; // __TEXT is non-writable | | |
| 5593 | } | | |
| 5594 | | 4955 | |
| 5595 | const seg = self.getSegment(match); | 4956 | log.debug("dyld info for {s},{s}", .{ header.segName(), header.sectName() }); |
| 5596 | const sect = self.getSection(match); | | |
| 5597 | log.debug("dyld info for {s},{s}", .{ sect.segName(), sect.sectName() }); | | |
| 5598 | | | |
| 5599 | while (true) { | | |
| 5600 | log.debug(" ATOM(%{d}, '{s}')", .{ atom.sym_index, atom.getName(self) }); | | |
| 5601 | const sym = atom.getSymbol(self); | | |
| 5602 | const base_offset = sym.n_value - seg.inner.vmaddr; | | |
| 5603 | | | |
| 5604 | for (atom.rebases.items) |offset| { | | |
| 5605 | log.debug(" | rebase at {x}", .{base_offset + offset}); | | |
| 5606 | try rebase_pointers.append(.{ | | |
| 5607 | .offset = base_offset + offset, | | |
| 5608 | .segment_id = match.seg, | | |
| 5609 | }); | | |
| 5610 | } | | |
| 5611 | | 4957 | |
| 5612 | for (atom.bindings.items) |binding| { | 4958 | const seg = self.segments.items[segment_index]; |
| 5613 | const bind_sym = self.getSymbol(binding.target); | | |
| 5614 | const bind_sym_name = self.getSymbolName(binding.target); | | |
| 5615 | const dylib_ordinal = @divTrunc( | | |
| 5616 | @bitCast(i16, bind_sym.n_desc), | | |
| 5617 | macho.N_SYMBOL_RESOLVER, | | |
| 5618 | ); | | |
| 5619 | var flags: u4 = 0; | | |
| 5620 | log.debug(" | bind at {x}, import('{s}') in dylib({d})", .{ | | |
| 5621 | binding.offset + base_offset, | | |
| 5622 | bind_sym_name, | | |
| 5623 | dylib_ordinal, | | |
| 5624 | }); | | |
| 5625 | if (bind_sym.weakRef()) { | | |
| 5626 | log.debug(" | marking as weak ref ", .{}); | | |
| 5627 | flags |= @truncate(u4, macho.BIND_SYMBOL_FLAGS_WEAK_IMPORT); | | |
| 5628 | } | | |
| 5629 | try bind_pointers.append(.{ | | |
| 5630 | .offset = binding.offset + base_offset, | | |
| 5631 | .segment_id = match.seg, | | |
| 5632 | .dylib_ordinal = dylib_ordinal, | | |
| 5633 | .name = bind_sym_name, | | |
| 5634 | .bind_flags = flags, | | |
| 5635 | }); | | |
| 5636 | } | | |
| 5637 | | 4959 | |
| 5638 | for (atom.lazy_bindings.items) |binding| { | 4960 | while (true) { |
| 5639 | const bind_sym = self.getSymbol(binding.target); | 4961 | log.debug(" ATOM(%{d}, '{s}')", .{ atom.sym_index, atom.getName(self) }); |
| 5640 | const bind_sym_name = self.getSymbolName(binding.target); | 4962 | const sym = atom.getSymbol(self); |
| 5641 | const dylib_ordinal = @divTrunc( | 4963 | const base_offset = sym.n_value - seg.vmaddr; |
| 5642 | @bitCast(i16, bind_sym.n_desc), | 4964 | |
| 5643 | macho.N_SYMBOL_RESOLVER, | 4965 | for (atom.rebases.items) |offset| { |
| 5644 | ); | 4966 | log.debug(" | rebase at {x}", .{base_offset + offset}); |
| 5645 | var flags: u4 = 0; | 4967 | try rebase_pointers.append(.{ |
| 5646 | log.debug(" | lazy bind at {x} import('{s}') ord({d})", .{ | 4968 | .offset = base_offset + offset, |
| 5647 | binding.offset + base_offset, | 4969 | .segment_id = segment_index, |
| 5648 | bind_sym_name, | 4970 | }); |
| 5649 | dylib_ordinal, | 4971 | } |
| 5650 | }); | 4972 | |
| 5651 | if (bind_sym.weakRef()) { | 4973 | for (atom.bindings.items) |binding| { |
| 5652 | log.debug(" | marking as weak ref ", .{}); | 4974 | const bind_sym = self.getSymbol(binding.target); |
| 5653 | flags |= @truncate(u4, macho.BIND_SYMBOL_FLAGS_WEAK_IMPORT); | 4975 | const bind_sym_name = self.getSymbolName(binding.target); |
| 5654 | } | 4976 | const dylib_ordinal = @divTrunc( |
| 5655 | try lazy_bind_pointers.append(.{ | 4977 | @bitCast(i16, bind_sym.n_desc), |
| 5656 | .offset = binding.offset + base_offset, | 4978 | macho.N_SYMBOL_RESOLVER, |
| 5657 | .segment_id = match.seg, | 4979 | ); |
| 5658 | .dylib_ordinal = dylib_ordinal, | 4980 | var flags: u4 = 0; |
| 5659 | .name = bind_sym_name, | 4981 | log.debug(" | bind at {x}, import('{s}') in dylib({d})", .{ |
| 5660 | .bind_flags = flags, | 4982 | binding.offset + base_offset, |
| 5661 | }); | 4983 | bind_sym_name, |
| | 4984 | dylib_ordinal, |
| | 4985 | }); |
| | 4986 | if (bind_sym.weakRef()) { |
| | 4987 | log.debug(" | marking as weak ref ", .{}); |
| | 4988 | flags |= @truncate(u4, macho.BIND_SYMBOL_FLAGS_WEAK_IMPORT); |
| 5662 | } | 4989 | } |
| | 4990 | try bind_pointers.append(.{ |
| | 4991 | .offset = binding.offset + base_offset, |
| | 4992 | .segment_id = segment_index, |
| | 4993 | .dylib_ordinal = dylib_ordinal, |
| | 4994 | .name = bind_sym_name, |
| | 4995 | .bind_flags = flags, |
| | 4996 | }); |
| | 4997 | } |
| 5663 | | 4998 | |
| 5664 | if (atom.prev) |prev| { | 4999 | for (atom.lazy_bindings.items) |binding| { |
| 5665 | atom = prev; | 5000 | const bind_sym = self.getSymbol(binding.target); |
| 5666 | } else break; | 5001 | const bind_sym_name = self.getSymbolName(binding.target); |
| | 5002 | const dylib_ordinal = @divTrunc( |
| | 5003 | @bitCast(i16, bind_sym.n_desc), |
| | 5004 | macho.N_SYMBOL_RESOLVER, |
| | 5005 | ); |
| | 5006 | var flags: u4 = 0; |
| | 5007 | log.debug(" | lazy bind at {x} import('{s}') ord({d})", .{ |
| | 5008 | binding.offset + base_offset, |
| | 5009 | bind_sym_name, |
| | 5010 | dylib_ordinal, |
| | 5011 | }); |
| | 5012 | if (bind_sym.weakRef()) { |
| | 5013 | log.debug(" | marking as weak ref ", .{}); |
| | 5014 | flags |= @truncate(u4, macho.BIND_SYMBOL_FLAGS_WEAK_IMPORT); |
| | 5015 | } |
| | 5016 | try lazy_bind_pointers.append(.{ |
| | 5017 | .offset = binding.offset + base_offset, |
| | 5018 | .segment_id = segment_index, |
| | 5019 | .dylib_ordinal = dylib_ordinal, |
| | 5020 | .name = bind_sym_name, |
| | 5021 | .bind_flags = flags, |
| | 5022 | }); |
| 5667 | } | 5023 | } |
| | 5024 | |
| | 5025 | if (atom.prev) |prev| { |
| | 5026 | atom = prev; |
| | 5027 | } else break; |
| 5668 | } | 5028 | } |
| 5669 | } | 5029 | } |
| 5670 | | 5030 | |
| ... | @@ -5675,8 +5035,8 @@ fn writeDyldInfoData(self: *MachO) !void { | ... | @@ -5675,8 +5035,8 @@ fn writeDyldInfoData(self: *MachO) !void { |
| 5675 | // TODO handle macho.EXPORT_SYMBOL_FLAGS_REEXPORT and macho.EXPORT_SYMBOL_FLAGS_STUB_AND_RESOLVER. | 5035 | // TODO handle macho.EXPORT_SYMBOL_FLAGS_REEXPORT and macho.EXPORT_SYMBOL_FLAGS_STUB_AND_RESOLVER. |
| 5676 | log.debug("generating export trie", .{}); | 5036 | log.debug("generating export trie", .{}); |
| 5677 | | 5037 | |
| 5678 | const text_segment = self.load_commands.items[self.text_segment_cmd_index.?].segment; | 5038 | const text_segment = self.segments.items[self.text_segment_cmd_index.?]; |
| 5679 | const base_address = text_segment.inner.vmaddr; | 5039 | const base_address = text_segment.vmaddr; |
| 5680 | | 5040 | |
| 5681 | if (self.base.options.output_mode == .Exe) { | 5041 | if (self.base.options.output_mode == .Exe) { |
| 5682 | for (&[_]SymbolWithLoc{ | 5042 | for (&[_]SymbolWithLoc{ |
| ... | @@ -5714,103 +5074,91 @@ fn writeDyldInfoData(self: *MachO) !void { | ... | @@ -5714,103 +5074,91 @@ fn writeDyldInfoData(self: *MachO) !void { |
| 5714 | try trie.finalize(gpa); | 5074 | try trie.finalize(gpa); |
| 5715 | } | 5075 | } |
| 5716 | | 5076 | |
| 5717 | const seg = &self.load_commands.items[self.linkedit_segment_cmd_index.?].segment; | 5077 | const link_seg = &self.segments.items[self.linkedit_segment_cmd_index.?]; |
| 5718 | const dyld_info = &self.load_commands.items[self.dyld_info_cmd_index.?].dyld_info_only; | 5078 | const rebase_off = mem.alignForwardGeneric(u64, link_seg.fileoff, @alignOf(u64)); |
| 5719 | | 5079 | assert(rebase_off == link_seg.fileoff); |
| 5720 | const rebase_off = mem.alignForwardGeneric(u64, seg.inner.fileoff, @alignOf(u64)); | | |
| 5721 | const rebase_size = try bind.rebaseInfoSize(rebase_pointers.items); | 5080 | const rebase_size = try bind.rebaseInfoSize(rebase_pointers.items); |
| 5722 | dyld_info.rebase_off = @intCast(u32, rebase_off); | 5081 | log.debug("writing rebase info from 0x{x} to 0x{x}", .{ rebase_off, rebase_off + rebase_size }); |
| 5723 | dyld_info.rebase_size = @intCast(u32, rebase_size); | | |
| 5724 | log.debug("writing rebase info from 0x{x} to 0x{x}", .{ | | |
| 5725 | dyld_info.rebase_off, | | |
| 5726 | dyld_info.rebase_off + dyld_info.rebase_size, | | |
| 5727 | }); | | |
| 5728 | | 5082 | |
| 5729 | const bind_off = mem.alignForwardGeneric(u64, dyld_info.rebase_off + dyld_info.rebase_size, @alignOf(u64)); | 5083 | const bind_off = mem.alignForwardGeneric(u64, rebase_off + rebase_size, @alignOf(u64)); |
| 5730 | const bind_size = try bind.bindInfoSize(bind_pointers.items); | 5084 | const bind_size = try bind.bindInfoSize(bind_pointers.items); |
| 5731 | dyld_info.bind_off = @intCast(u32, bind_off); | 5085 | log.debug("writing bind info from 0x{x} to 0x{x}", .{ bind_off, bind_off + bind_size }); |
| 5732 | dyld_info.bind_size = @intCast(u32, bind_size); | | |
| 5733 | log.debug("writing bind info from 0x{x} to 0x{x}", .{ | | |
| 5734 | dyld_info.bind_off, | | |
| 5735 | dyld_info.bind_off + dyld_info.bind_size, | | |
| 5736 | }); | | |
| 5737 | | 5086 | |
| 5738 | const lazy_bind_off = mem.alignForwardGeneric(u64, dyld_info.bind_off + dyld_info.bind_size, @alignOf(u64)); | 5087 | const lazy_bind_off = mem.alignForwardGeneric(u64, bind_off + bind_size, @alignOf(u64)); |
| 5739 | const lazy_bind_size = try bind.lazyBindInfoSize(lazy_bind_pointers.items); | 5088 | const lazy_bind_size = try bind.lazyBindInfoSize(lazy_bind_pointers.items); |
| 5740 | dyld_info.lazy_bind_off = @intCast(u32, lazy_bind_off); | 5089 | log.debug("writing lazy bind info from 0x{x} to 0x{x}", .{ lazy_bind_off, lazy_bind_off + lazy_bind_size }); |
| 5741 | dyld_info.lazy_bind_size = @intCast(u32, lazy_bind_size); | | |
| 5742 | log.debug("writing lazy bind info from 0x{x} to 0x{x}", .{ | | |
| 5743 | dyld_info.lazy_bind_off, | | |
| 5744 | dyld_info.lazy_bind_off + dyld_info.lazy_bind_size, | | |
| 5745 | }); | | |
| 5746 | | 5090 | |
| 5747 | const export_off = mem.alignForwardGeneric(u64, dyld_info.lazy_bind_off + dyld_info.lazy_bind_size, @alignOf(u64)); | 5091 | const export_off = mem.alignForwardGeneric(u64, lazy_bind_off + lazy_bind_size, @alignOf(u64)); |
| 5748 | const export_size = trie.size; | 5092 | const export_size = trie.size; |
| 5749 | dyld_info.export_off = @intCast(u32, export_off); | 5093 | log.debug("writing export trie from 0x{x} to 0x{x}", .{ export_off, export_off + export_size }); |
| 5750 | dyld_info.export_size = @intCast(u32, export_size); | | |
| 5751 | log.debug("writing export trie from 0x{x} to 0x{x}", .{ | | |
| 5752 | dyld_info.export_off, | | |
| 5753 | dyld_info.export_off + dyld_info.export_size, | | |
| 5754 | }); | | |
| 5755 | | 5094 | |
| 5756 | seg.inner.filesize = dyld_info.export_off + dyld_info.export_size - seg.inner.fileoff; | 5095 | const needed_size = export_off + export_size - rebase_off; |
| | 5096 | link_seg.filesize = needed_size; |
| 5757 | | 5097 | |
| 5758 | const needed_size = dyld_info.export_off + dyld_info.export_size - dyld_info.rebase_off; | 5098 | var buffer = try gpa.alloc(u8, math.cast(usize, needed_size) orelse return error.Overflow); |
| 5759 | var buffer = try gpa.alloc(u8, needed_size); | | |
| 5760 | defer gpa.free(buffer); | 5099 | defer gpa.free(buffer); |
| 5761 | mem.set(u8, buffer, 0); | 5100 | mem.set(u8, buffer, 0); |
| 5762 | | 5101 | |
| 5763 | var stream = std.io.fixedBufferStream(buffer); | 5102 | var stream = std.io.fixedBufferStream(buffer); |
| 5764 | const writer = stream.writer(); | 5103 | const writer = stream.writer(); |
| 5765 | | 5104 | |
| 5766 | const base_off = dyld_info.rebase_off; | | |
| 5767 | try bind.writeRebaseInfo(rebase_pointers.items, writer); | 5105 | try bind.writeRebaseInfo(rebase_pointers.items, writer); |
| 5768 | try stream.seekTo(dyld_info.bind_off - base_off); | 5106 | try stream.seekTo(bind_off - rebase_off); |
| 5769 | | 5107 | |
| 5770 | try bind.writeBindInfo(bind_pointers.items, writer); | 5108 | try bind.writeBindInfo(bind_pointers.items, writer); |
| 5771 | try stream.seekTo(dyld_info.lazy_bind_off - base_off); | 5109 | try stream.seekTo(lazy_bind_off - rebase_off); |
| 5772 | | 5110 | |
| 5773 | try bind.writeLazyBindInfo(lazy_bind_pointers.items, writer); | 5111 | try bind.writeLazyBindInfo(lazy_bind_pointers.items, writer); |
| 5774 | try stream.seekTo(dyld_info.export_off - base_off); | 5112 | try stream.seekTo(export_off - rebase_off); |
| 5775 | | 5113 | |
| 5776 | _ = try trie.write(writer); | 5114 | _ = try trie.write(writer); |
| 5777 | | 5115 | |
| 5778 | log.debug("writing dyld info from 0x{x} to 0x{x}", .{ | 5116 | log.debug("writing dyld info from 0x{x} to 0x{x}", .{ |
| 5779 | dyld_info.rebase_off, | 5117 | rebase_off, |
| 5780 | dyld_info.rebase_off + needed_size, | 5118 | rebase_off + needed_size, |
| 5781 | }); | 5119 | }); |
| 5782 | | 5120 | |
| 5783 | try self.base.file.?.pwriteAll(buffer, dyld_info.rebase_off); | 5121 | try self.base.file.?.pwriteAll(buffer, rebase_off); |
| 5784 | try self.populateLazyBindOffsetsInStubHelper( | 5122 | const start = math.cast(usize, lazy_bind_off - rebase_off) orelse return error.Overflow; |
| 5785 | buffer[dyld_info.lazy_bind_off - base_off ..][0..dyld_info.lazy_bind_size], | 5123 | const end = start + (math.cast(usize, lazy_bind_size) orelse return error.Overflow); |
| 5786 | ); | 5124 | try self.populateLazyBindOffsetsInStubHelper(buffer[start..end]); |
| 5787 | | 5125 | |
| 5788 | self.load_commands_dirty = true; | 5126 | try lc_writer.writeStruct(macho.dyld_info_command{ |
| | 5127 | .cmd = .DYLD_INFO_ONLY, |
| | 5128 | .cmdsize = @sizeOf(macho.dyld_info_command), |
| | 5129 | .rebase_off = @intCast(u32, rebase_off), |
| | 5130 | .rebase_size = @intCast(u32, rebase_size), |
| | 5131 | .bind_off = @intCast(u32, bind_off), |
| | 5132 | .bind_size = @intCast(u32, bind_size), |
| | 5133 | .weak_bind_off = 0, |
| | 5134 | .weak_bind_size = 0, |
| | 5135 | .lazy_bind_off = @intCast(u32, lazy_bind_off), |
| | 5136 | .lazy_bind_size = @intCast(u32, lazy_bind_size), |
| | 5137 | .export_off = @intCast(u32, export_off), |
| | 5138 | .export_size = @intCast(u32, export_size), |
| | 5139 | }); |
| | 5140 | ncmds.* += 1; |
| 5789 | } | 5141 | } |
| 5790 | | 5142 | |
| 5791 | fn populateLazyBindOffsetsInStubHelper(self: *MachO, buffer: []const u8) !void { | 5143 | fn populateLazyBindOffsetsInStubHelper(self: *MachO, buffer: []const u8) !void { |
| 5792 | const gpa = self.base.allocator; | 5144 | const gpa = self.base.allocator; |
| 5793 | const text_segment_cmd_index = self.text_segment_cmd_index orelse return; | 5145 | |
| 5794 | const stub_helper_section_index = self.stub_helper_section_index orelse return; | 5146 | const stub_helper_section_index = self.stub_helper_section_index orelse return; |
| 5795 | const last_atom = self.atoms.get(.{ | | |
| 5796 | .seg = text_segment_cmd_index, | | |
| 5797 | .sect = stub_helper_section_index, | | |
| 5798 | }) orelse return; | | |
| 5799 | if (self.stub_helper_preamble_atom == null) return; | 5147 | if (self.stub_helper_preamble_atom == null) return; |
| 5800 | if (last_atom == self.stub_helper_preamble_atom.?) return; | 5148 | |
| | 5149 | const section = self.sections.get(stub_helper_section_index); |
| | 5150 | const last_atom = section.last_atom orelse return; |
| | 5151 | if (last_atom == self.stub_helper_preamble_atom.?) return; // TODO is this a redundant check? |
| 5801 | | 5152 | |
| 5802 | var table = std.AutoHashMap(i64, *Atom).init(gpa); | 5153 | var table = std.AutoHashMap(i64, *Atom).init(gpa); |
| 5803 | defer table.deinit(); | 5154 | defer table.deinit(); |
| 5804 | | 5155 | |
| 5805 | { | 5156 | { |
| 5806 | var stub_atom = last_atom; | 5157 | var stub_atom = last_atom; |
| 5807 | var laptr_atom = self.atoms.get(.{ | 5158 | var laptr_atom = self.sections.items(.last_atom)[self.la_symbol_ptr_section_index.?].?; |
| 5808 | .seg = self.data_segment_cmd_index.?, | | |
| 5809 | .sect = self.la_symbol_ptr_section_index.?, | | |
| 5810 | }).?; | | |
| 5811 | const base_addr = blk: { | 5159 | const base_addr = blk: { |
| 5812 | const seg = self.load_commands.items[self.data_segment_cmd_index.?].segment; | 5160 | const seg = self.segments.items[self.data_segment_cmd_index.?]; |
| 5813 | break :blk seg.inner.vmaddr; | 5161 | break :blk seg.vmaddr; |
| 5814 | }; | 5162 | }; |
| 5815 | | 5163 | |
| 5816 | while (true) { | 5164 | while (true) { |
| ... | @@ -5871,10 +5219,7 @@ fn populateLazyBindOffsetsInStubHelper(self: *MachO, buffer: []const u8) !void { | ... | @@ -5871,10 +5219,7 @@ fn populateLazyBindOffsetsInStubHelper(self: *MachO, buffer: []const u8) !void { |
| 5871 | } | 5219 | } |
| 5872 | } | 5220 | } |
| 5873 | | 5221 | |
| 5874 | const sect = self.getSection(.{ | 5222 | const header = self.sections.items(.header)[stub_helper_section_index]; |
| 5875 | .seg = text_segment_cmd_index, | | |
| 5876 | .sect = stub_helper_section_index, | | |
| 5877 | }); | | |
| 5878 | const stub_offset: u4 = switch (self.base.options.target.cpu.arch) { | 5223 | const stub_offset: u4 = switch (self.base.options.target.cpu.arch) { |
| 5879 | .x86_64 => 1, | 5224 | .x86_64 => 1, |
| 5880 | .aarch64 => 2 * @sizeOf(u32), | 5225 | .aarch64 => 2 * @sizeOf(u32), |
| ... | @@ -5886,7 +5231,7 @@ fn populateLazyBindOffsetsInStubHelper(self: *MachO, buffer: []const u8) !void { | ... | @@ -5886,7 +5231,7 @@ fn populateLazyBindOffsetsInStubHelper(self: *MachO, buffer: []const u8) !void { |
| 5886 | while (offsets.popOrNull()) |bind_offset| { | 5231 | while (offsets.popOrNull()) |bind_offset| { |
| 5887 | const atom = table.get(bind_offset.sym_offset).?; | 5232 | const atom = table.get(bind_offset.sym_offset).?; |
| 5888 | const sym = atom.getSymbol(self); | 5233 | const sym = atom.getSymbol(self); |
| 5889 | const file_offset = sect.offset + sym.n_value - sect.addr + stub_offset; | 5234 | const file_offset = header.offset + sym.n_value - header.addr + stub_offset; |
| 5890 | mem.writeIntLittle(u32, &buf, bind_offset.offset); | 5235 | mem.writeIntLittle(u32, &buf, bind_offset.offset); |
| 5891 | log.debug("writing lazy bind offset in stub helper of 0x{x} for symbol {s} at offset 0x{x}", .{ | 5236 | log.debug("writing lazy bind offset in stub helper of 0x{x} for symbol {s} at offset 0x{x}", .{ |
| 5892 | bind_offset.offset, | 5237 | bind_offset.offset, |
| ... | @@ -5899,14 +5244,14 @@ fn populateLazyBindOffsetsInStubHelper(self: *MachO, buffer: []const u8) !void { | ... | @@ -5899,14 +5244,14 @@ fn populateLazyBindOffsetsInStubHelper(self: *MachO, buffer: []const u8) !void { |
| 5899 | | 5244 | |
| 5900 | const asc_u64 = std.sort.asc(u64); | 5245 | const asc_u64 = std.sort.asc(u64); |
| 5901 | | 5246 | |
| 5902 | fn writeFunctionStarts(self: *MachO) !void { | 5247 | fn writeFunctionStarts(self: *MachO, ncmds: *u32, lc_writer: anytype) !void { |
| 5903 | const text_seg_index = self.text_segment_cmd_index orelse return; | | |
| 5904 | const text_sect_index = self.text_section_index orelse return; | | |
| 5905 | const text_seg = self.load_commands.items[text_seg_index].segment; | | |
| 5906 | | | |
| 5907 | const tracy = trace(@src()); | 5248 | const tracy = trace(@src()); |
| 5908 | defer tracy.end(); | 5249 | defer tracy.end(); |
| 5909 | | 5250 | |
| | 5251 | const text_seg_index = self.text_segment_cmd_index orelse return; |
| | 5252 | const text_sect_index = self.text_section_index orelse return; |
| | 5253 | const text_seg = self.segments.items[text_seg_index]; |
| | 5254 | |
| 5910 | const gpa = self.base.allocator; | 5255 | const gpa = self.base.allocator; |
| 5911 | | 5256 | |
| 5912 | // We need to sort by address first | 5257 | // We need to sort by address first |
| ... | @@ -5918,8 +5263,8 @@ fn writeFunctionStarts(self: *MachO) !void { | ... | @@ -5918,8 +5263,8 @@ fn writeFunctionStarts(self: *MachO) !void { |
| 5918 | const sym = self.getSymbol(global); | 5263 | const sym = self.getSymbol(global); |
| 5919 | if (sym.undf()) continue; | 5264 | if (sym.undf()) continue; |
| 5920 | if (sym.n_desc == N_DESC_GCED) continue; | 5265 | if (sym.n_desc == N_DESC_GCED) continue; |
| 5921 | const match = self.getMatchingSectionFromOrdinal(sym.n_sect); | 5266 | const sect_id = sym.n_sect - 1; |
| 5922 | if (match.seg != text_seg_index or match.sect != text_sect_index) continue; | 5267 | if (sect_id != text_sect_index) continue; |
| 5923 | | 5268 | |
| 5924 | addresses.appendAssumeCapacity(sym.n_value); | 5269 | addresses.appendAssumeCapacity(sym.n_value); |
| 5925 | } | 5270 | } |
| ... | @@ -5932,7 +5277,7 @@ fn writeFunctionStarts(self: *MachO) !void { | ... | @@ -5932,7 +5277,7 @@ fn writeFunctionStarts(self: *MachO) !void { |
| 5932 | | 5277 | |
| 5933 | var last_off: u32 = 0; | 5278 | var last_off: u32 = 0; |
| 5934 | for (addresses.items) |addr| { | 5279 | for (addresses.items) |addr| { |
| 5935 | const offset = @intCast(u32, addr - text_seg.inner.vmaddr); | 5280 | const offset = @intCast(u32, addr - text_seg.vmaddr); |
| 5936 | const diff = offset - last_off; | 5281 | const diff = offset - last_off; |
| 5937 | | 5282 | |
| 5938 | if (diff == 0) continue; | 5283 | if (diff == 0) continue; |
| ... | @@ -5951,22 +5296,22 @@ fn writeFunctionStarts(self: *MachO) !void { | ... | @@ -5951,22 +5296,22 @@ fn writeFunctionStarts(self: *MachO) !void { |
| 5951 | try std.leb.writeULEB128(buffer.writer(), offset); | 5296 | try std.leb.writeULEB128(buffer.writer(), offset); |
| 5952 | } | 5297 | } |
| 5953 | | 5298 | |
| 5954 | const seg = &self.load_commands.items[self.linkedit_segment_cmd_index.?].segment; | 5299 | const link_seg = &self.segments.items[self.linkedit_segment_cmd_index.?]; |
| 5955 | const fn_cmd = &self.load_commands.items[self.function_starts_cmd_index.?].linkedit_data; | 5300 | const offset = mem.alignForwardGeneric(u64, link_seg.fileoff + link_seg.filesize, @alignOf(u64)); |
| | 5301 | const needed_size = buffer.items.len; |
| | 5302 | link_seg.filesize = offset + needed_size - link_seg.fileoff; |
| 5956 | | 5303 | |
| 5957 | const dataoff = mem.alignForwardGeneric(u64, seg.inner.fileoff + seg.inner.filesize, @alignOf(u64)); | 5304 | log.debug("writing function starts info from 0x{x} to 0x{x}", .{ offset, offset + needed_size }); |
| 5958 | const datasize = buffer.items.len; | | |
| 5959 | fn_cmd.dataoff = @intCast(u32, dataoff); | | |
| 5960 | fn_cmd.datasize = @intCast(u32, datasize); | | |
| 5961 | seg.inner.filesize = fn_cmd.dataoff + fn_cmd.datasize - seg.inner.fileoff; | | |
| 5962 | | 5305 | |
| 5963 | log.debug("writing function starts info from 0x{x} to 0x{x}", .{ | 5306 | try self.base.file.?.pwriteAll(buffer.items, offset); |
| 5964 | fn_cmd.dataoff, | | |
| 5965 | fn_cmd.dataoff + fn_cmd.datasize, | | |
| 5966 | }); | | |
| 5967 | | 5307 | |
| 5968 | try self.base.file.?.pwriteAll(buffer.items, fn_cmd.dataoff); | 5308 | try lc_writer.writeStruct(macho.linkedit_data_command{ |
| 5969 | self.load_commands_dirty = true; | 5309 | .cmd = .FUNCTION_STARTS, |
| | 5310 | .cmdsize = @sizeOf(macho.linkedit_data_command), |
| | 5311 | .dataoff = @intCast(u32, offset), |
| | 5312 | .datasize = @intCast(u32, needed_size), |
| | 5313 | }); |
| | 5314 | ncmds.* += 1; |
| 5970 | } | 5315 | } |
| 5971 | | 5316 | |
| 5972 | fn filterDataInCode( | 5317 | fn filterDataInCode( |
| ... | @@ -5988,17 +5333,15 @@ fn filterDataInCode( | ... | @@ -5988,17 +5333,15 @@ fn filterDataInCode( |
| 5988 | return dices[start..end]; | 5333 | return dices[start..end]; |
| 5989 | } | 5334 | } |
| 5990 | | 5335 | |
| 5991 | fn writeDataInCode(self: *MachO) !void { | 5336 | fn writeDataInCode(self: *MachO, ncmds: *u32, lc_writer: anytype) !void { |
| 5992 | const tracy = trace(@src()); | 5337 | const tracy = trace(@src()); |
| 5993 | defer tracy.end(); | 5338 | defer tracy.end(); |
| 5994 | | 5339 | |
| 5995 | var out_dice = std.ArrayList(macho.data_in_code_entry).init(self.base.allocator); | 5340 | var out_dice = std.ArrayList(macho.data_in_code_entry).init(self.base.allocator); |
| 5996 | defer out_dice.deinit(); | 5341 | defer out_dice.deinit(); |
| 5997 | | 5342 | |
| 5998 | const text_sect = self.getSection(.{ | 5343 | const text_sect_id = self.text_section_index orelse return; |
| 5999 | .seg = self.text_segment_cmd_index orelse return, | 5344 | const text_sect_header = self.sections.items(.header)[text_sect_id]; |
| 6000 | .sect = self.text_section_index orelse return, | | |
| 6001 | }); | | |
| 6002 | | 5345 | |
| 6003 | for (self.objects.items) |object| { | 5346 | for (self.objects.items) |object| { |
| 6004 | const dice = object.parseDataInCode() orelse continue; | 5347 | const dice = object.parseDataInCode() orelse continue; |
| ... | @@ -6008,15 +5351,15 @@ fn writeDataInCode(self: *MachO) !void { | ... | @@ -6008,15 +5351,15 @@ fn writeDataInCode(self: *MachO) !void { |
| 6008 | const sym = atom.getSymbol(self); | 5351 | const sym = atom.getSymbol(self); |
| 6009 | if (sym.n_desc == N_DESC_GCED) continue; | 5352 | if (sym.n_desc == N_DESC_GCED) continue; |
| 6010 | | 5353 | |
| 6011 | const match = self.getMatchingSectionFromOrdinal(sym.n_sect); | 5354 | const sect_id = sym.n_sect - 1; |
| 6012 | if (match.seg != self.text_segment_cmd_index.? and match.sect != self.text_section_index.?) { | 5355 | if (sect_id != self.text_section_index.?) { |
| 6013 | continue; | 5356 | continue; |
| 6014 | } | 5357 | } |
| 6015 | | 5358 | |
| 6016 | const source_sym = object.getSourceSymbol(atom.sym_index) orelse continue; | 5359 | const source_sym = object.getSourceSymbol(atom.sym_index) orelse continue; |
| 6017 | const source_addr = math.cast(u32, source_sym.n_value) orelse return error.Overflow; | 5360 | const source_addr = math.cast(u32, source_sym.n_value) orelse return error.Overflow; |
| 6018 | const filtered_dice = filterDataInCode(dice, source_addr, source_addr + atom.size); | 5361 | const filtered_dice = filterDataInCode(dice, source_addr, source_addr + atom.size); |
| 6019 | const base = math.cast(u32, sym.n_value - text_sect.addr + text_sect.offset) orelse | 5362 | const base = math.cast(u32, sym.n_value - text_sect_header.addr + text_sect_header.offset) orelse |
| 6020 | return error.Overflow; | 5363 | return error.Overflow; |
| 6021 | | 5364 | |
| 6022 | for (filtered_dice) |single| { | 5365 | for (filtered_dice) |single| { |
| ... | @@ -6030,33 +5373,63 @@ fn writeDataInCode(self: *MachO) !void { | ... | @@ -6030,33 +5373,63 @@ fn writeDataInCode(self: *MachO) !void { |
| 6030 | } | 5373 | } |
| 6031 | } | 5374 | } |
| 6032 | | 5375 | |
| 6033 | const seg = &self.load_commands.items[self.linkedit_segment_cmd_index.?].segment; | 5376 | const seg = &self.segments.items[self.linkedit_segment_cmd_index.?]; |
| 6034 | const dice_cmd = &self.load_commands.items[self.data_in_code_cmd_index.?].linkedit_data; | 5377 | const offset = mem.alignForwardGeneric(u64, seg.fileoff + seg.filesize, @alignOf(u64)); |
| | 5378 | const needed_size = out_dice.items.len * @sizeOf(macho.data_in_code_entry); |
| | 5379 | seg.filesize = offset + needed_size - seg.fileoff; |
| 6035 | | 5380 | |
| 6036 | const dataoff = mem.alignForwardGeneric(u64, seg.inner.fileoff + seg.inner.filesize, @alignOf(u64)); | 5381 | log.debug("writing data-in-code from 0x{x} to 0x{x}", .{ offset, offset + needed_size }); |
| 6037 | const datasize = out_dice.items.len * @sizeOf(macho.data_in_code_entry); | | |
| 6038 | dice_cmd.dataoff = @intCast(u32, dataoff); | | |
| 6039 | dice_cmd.datasize = @intCast(u32, datasize); | | |
| 6040 | seg.inner.filesize = dice_cmd.dataoff + dice_cmd.datasize - seg.inner.fileoff; | | |
| 6041 | | 5382 | |
| 6042 | log.debug("writing data-in-code from 0x{x} to 0x{x}", .{ | 5383 | try self.base.file.?.pwriteAll(mem.sliceAsBytes(out_dice.items), offset); |
| 6043 | dice_cmd.dataoff, | 5384 | try lc_writer.writeStruct(macho.linkedit_data_command{ |
| 6044 | dice_cmd.dataoff + dice_cmd.datasize, | 5385 | .cmd = .DATA_IN_CODE, |
| | 5386 | .cmdsize = @sizeOf(macho.linkedit_data_command), |
| | 5387 | .dataoff = @intCast(u32, offset), |
| | 5388 | .datasize = @intCast(u32, needed_size), |
| 6045 | }); | 5389 | }); |
| 6046 | | 5390 | ncmds.* += 1; |
| 6047 | try self.base.file.?.pwriteAll(mem.sliceAsBytes(out_dice.items), dice_cmd.dataoff); | | |
| 6048 | self.load_commands_dirty = true; | | |
| 6049 | } | 5391 | } |
| 6050 | | 5392 | |
| 6051 | fn writeSymtab(self: *MachO) !void { | 5393 | fn writeSymtabs(self: *MachO, ncmds: *u32, lc_writer: anytype) !void { |
| 6052 | const tracy = trace(@src()); | 5394 | var symtab_cmd = macho.symtab_command{ |
| 6053 | defer tracy.end(); | 5395 | .cmdsize = @sizeOf(macho.symtab_command), |
| | 5396 | .symoff = 0, |
| | 5397 | .nsyms = 0, |
| | 5398 | .stroff = 0, |
| | 5399 | .strsize = 0, |
| | 5400 | }; |
| | 5401 | var dysymtab_cmd = macho.dysymtab_command{ |
| | 5402 | .cmdsize = @sizeOf(macho.dysymtab_command), |
| | 5403 | .ilocalsym = 0, |
| | 5404 | .nlocalsym = 0, |
| | 5405 | .iextdefsym = 0, |
| | 5406 | .nextdefsym = 0, |
| | 5407 | .iundefsym = 0, |
| | 5408 | .nundefsym = 0, |
| | 5409 | .tocoff = 0, |
| | 5410 | .ntoc = 0, |
| | 5411 | .modtaboff = 0, |
| | 5412 | .nmodtab = 0, |
| | 5413 | .extrefsymoff = 0, |
| | 5414 | .nextrefsyms = 0, |
| | 5415 | .indirectsymoff = 0, |
| | 5416 | .nindirectsyms = 0, |
| | 5417 | .extreloff = 0, |
| | 5418 | .nextrel = 0, |
| | 5419 | .locreloff = 0, |
| | 5420 | .nlocrel = 0, |
| | 5421 | }; |
| | 5422 | var ctx = try self.writeSymtab(&symtab_cmd); |
| | 5423 | defer ctx.imports_table.deinit(); |
| | 5424 | try self.writeDysymtab(ctx, &dysymtab_cmd); |
| | 5425 | try self.writeStrtab(&symtab_cmd); |
| | 5426 | try lc_writer.writeStruct(symtab_cmd); |
| | 5427 | try lc_writer.writeStruct(dysymtab_cmd); |
| | 5428 | ncmds.* += 2; |
| | 5429 | } |
| 6054 | | 5430 | |
| | 5431 | fn writeSymtab(self: *MachO, lc: *macho.symtab_command) !SymtabCtx { |
| 6055 | const gpa = self.base.allocator; | 5432 | const gpa = self.base.allocator; |
| 6056 | const seg = &self.load_commands.items[self.linkedit_segment_cmd_index.?].segment; | | |
| 6057 | const symtab = &self.load_commands.items[self.symtab_cmd_index.?].symtab; | | |
| 6058 | const symoff = mem.alignForwardGeneric(u64, seg.inner.fileoff + seg.inner.filesize, @alignOf(macho.nlist_64)); | | |
| 6059 | symtab.symoff = @intCast(u32, symoff); | | |
| 6060 | | 5433 | |
| 6061 | var locals = std.ArrayList(macho.nlist_64).init(gpa); | 5434 | var locals = std.ArrayList(macho.nlist_64).init(gpa); |
| 6062 | defer locals.deinit(); | 5435 | defer locals.deinit(); |
| ... | @@ -6101,8 +5474,8 @@ fn writeSymtab(self: *MachO) !void { | ... | @@ -6101,8 +5474,8 @@ fn writeSymtab(self: *MachO) !void { |
| 6101 | | 5474 | |
| 6102 | var imports = std.ArrayList(macho.nlist_64).init(gpa); | 5475 | var imports = std.ArrayList(macho.nlist_64).init(gpa); |
| 6103 | defer imports.deinit(); | 5476 | defer imports.deinit(); |
| | 5477 | |
| 6104 | var imports_table = std.AutoHashMap(SymbolWithLoc, u32).init(gpa); | 5478 | var imports_table = std.AutoHashMap(SymbolWithLoc, u32).init(gpa); |
| 6105 | defer imports_table.deinit(); | | |
| 6106 | | 5479 | |
| 6107 | for (self.globals.values()) |global| { | 5480 | for (self.globals.values()) |global| { |
| 6108 | const sym = self.getSymbol(global); | 5481 | const sym = self.getSymbol(global); |
| ... | @@ -6115,56 +5488,84 @@ fn writeSymtab(self: *MachO) !void { | ... | @@ -6115,56 +5488,84 @@ fn writeSymtab(self: *MachO) !void { |
| 6115 | try imports_table.putNoClobber(global, new_index); | 5488 | try imports_table.putNoClobber(global, new_index); |
| 6116 | } | 5489 | } |
| 6117 | | 5490 | |
| 6118 | const nlocals = locals.items.len; | 5491 | const nlocals = @intCast(u32, locals.items.len); |
| 6119 | const nexports = exports.items.len; | 5492 | const nexports = @intCast(u32, exports.items.len); |
| 6120 | const nimports = imports.items.len; | 5493 | const nimports = @intCast(u32, imports.items.len); |
| 6121 | symtab.nsyms = @intCast(u32, nlocals + nexports + nimports); | 5494 | const nsyms = nlocals + nexports + nimports; |
| | 5495 | |
| | 5496 | const seg = &self.segments.items[self.linkedit_segment_cmd_index.?]; |
| | 5497 | const offset = mem.alignForwardGeneric( |
| | 5498 | u64, |
| | 5499 | seg.fileoff + seg.filesize, |
| | 5500 | @alignOf(macho.nlist_64), |
| | 5501 | ); |
| | 5502 | const needed_size = nsyms * @sizeOf(macho.nlist_64); |
| | 5503 | seg.filesize = offset + needed_size - seg.fileoff; |
| 6122 | | 5504 | |
| 6123 | var buffer = std.ArrayList(u8).init(gpa); | 5505 | var buffer = std.ArrayList(u8).init(gpa); |
| 6124 | defer buffer.deinit(); | 5506 | defer buffer.deinit(); |
| 6125 | try buffer.ensureTotalCapacityPrecise(symtab.nsyms * @sizeOf(macho.nlist_64)); | 5507 | try buffer.ensureTotalCapacityPrecise(needed_size); |
| 6126 | buffer.appendSliceAssumeCapacity(mem.sliceAsBytes(locals.items)); | 5508 | buffer.appendSliceAssumeCapacity(mem.sliceAsBytes(locals.items)); |
| 6127 | buffer.appendSliceAssumeCapacity(mem.sliceAsBytes(exports.items)); | 5509 | buffer.appendSliceAssumeCapacity(mem.sliceAsBytes(exports.items)); |
| 6128 | buffer.appendSliceAssumeCapacity(mem.sliceAsBytes(imports.items)); | 5510 | buffer.appendSliceAssumeCapacity(mem.sliceAsBytes(imports.items)); |
| 6129 | | 5511 | |
| 6130 | log.debug("writing symtab from 0x{x} to 0x{x}", .{ symtab.symoff, symtab.symoff + buffer.items.len }); | 5512 | log.debug("writing symtab from 0x{x} to 0x{x}", .{ offset, offset + needed_size }); |
| 6131 | try self.base.file.?.pwriteAll(buffer.items, symtab.symoff); | 5513 | try self.base.file.?.pwriteAll(buffer.items, offset); |
| | 5514 | |
| | 5515 | lc.symoff = @intCast(u32, offset); |
| | 5516 | lc.nsyms = nsyms; |
| 6132 | | 5517 | |
| 6133 | seg.inner.filesize = symtab.symoff + buffer.items.len - seg.inner.fileoff; | 5518 | return SymtabCtx{ |
| | 5519 | .nlocalsym = nlocals, |
| | 5520 | .nextdefsym = nexports, |
| | 5521 | .nundefsym = nimports, |
| | 5522 | .imports_table = imports_table, |
| | 5523 | }; |
| | 5524 | } |
| | 5525 | |
| | 5526 | fn writeStrtab(self: *MachO, lc: *macho.symtab_command) !void { |
| | 5527 | const seg = &self.segments.items[self.linkedit_segment_cmd_index.?]; |
| | 5528 | const offset = mem.alignForwardGeneric(u64, seg.fileoff + seg.filesize, @alignOf(u64)); |
| | 5529 | const needed_size = self.strtab.buffer.items.len; |
| | 5530 | seg.filesize = offset + needed_size - seg.fileoff; |
| | 5531 | |
| | 5532 | log.debug("writing string table from 0x{x} to 0x{x}", .{ offset, offset + needed_size }); |
| 6134 | | 5533 | |
| 6135 | // Update dynamic symbol table. | 5534 | try self.base.file.?.pwriteAll(self.strtab.buffer.items, offset); |
| 6136 | const dysymtab = &self.load_commands.items[self.dysymtab_cmd_index.?].dysymtab; | 5535 | |
| 6137 | dysymtab.nlocalsym = @intCast(u32, nlocals); | 5536 | lc.stroff = @intCast(u32, offset); |
| 6138 | dysymtab.iextdefsym = dysymtab.nlocalsym; | 5537 | lc.strsize = @intCast(u32, needed_size); |
| 6139 | dysymtab.nextdefsym = @intCast(u32, nexports); | 5538 | } |
| 6140 | dysymtab.iundefsym = dysymtab.nlocalsym + dysymtab.nextdefsym; | 5539 | |
| 6141 | dysymtab.nundefsym = @intCast(u32, nimports); | 5540 | const SymtabCtx = struct { |
| | 5541 | nlocalsym: u32, |
| | 5542 | nextdefsym: u32, |
| | 5543 | nundefsym: u32, |
| | 5544 | imports_table: std.AutoHashMap(SymbolWithLoc, u32), |
| | 5545 | }; |
| 6142 | | 5546 | |
| | 5547 | fn writeDysymtab(self: *MachO, ctx: SymtabCtx, lc: *macho.dysymtab_command) !void { |
| | 5548 | const gpa = self.base.allocator; |
| 6143 | const nstubs = @intCast(u32, self.stubs_table.count()); | 5549 | const nstubs = @intCast(u32, self.stubs_table.count()); |
| 6144 | const ngot_entries = @intCast(u32, self.got_entries_table.count()); | 5550 | const ngot_entries = @intCast(u32, self.got_entries_table.count()); |
| | 5551 | const nindirectsyms = nstubs * 2 + ngot_entries; |
| | 5552 | const iextdefsym = ctx.nlocalsym; |
| | 5553 | const iundefsym = iextdefsym + ctx.nextdefsym; |
| 6145 | | 5554 | |
| 6146 | const indirectsymoff = mem.alignForwardGeneric(u64, seg.inner.fileoff + seg.inner.filesize, @alignOf(u64)); | 5555 | const seg = &self.segments.items[self.linkedit_segment_cmd_index.?]; |
| 6147 | dysymtab.indirectsymoff = @intCast(u32, indirectsymoff); | 5556 | const offset = mem.alignForwardGeneric(u64, seg.fileoff + seg.filesize, @alignOf(u64)); |
| 6148 | dysymtab.nindirectsyms = nstubs * 2 + ngot_entries; | 5557 | const needed_size = nindirectsyms * @sizeOf(u32); |
| 6149 | | 5558 | seg.filesize = offset + needed_size - seg.fileoff; |
| 6150 | seg.inner.filesize = dysymtab.indirectsymoff + dysymtab.nindirectsyms * @sizeOf(u32) - seg.inner.fileoff; | | |
| 6151 | | 5559 | |
| 6152 | log.debug("writing indirect symbol table from 0x{x} to 0x{x}", .{ | 5560 | log.debug("writing indirect symbol table from 0x{x} to 0x{x}", .{ offset, offset + needed_size }); |
| 6153 | dysymtab.indirectsymoff, | | |
| 6154 | dysymtab.indirectsymoff + dysymtab.nindirectsyms * @sizeOf(u32), | | |
| 6155 | }); | | |
| 6156 | | 5561 | |
| 6157 | var buf = std.ArrayList(u8).init(gpa); | 5562 | var buf = std.ArrayList(u8).init(gpa); |
| 6158 | defer buf.deinit(); | 5563 | defer buf.deinit(); |
| 6159 | try buf.ensureTotalCapacity(dysymtab.nindirectsyms * @sizeOf(u32)); | 5564 | try buf.ensureTotalCapacity(needed_size); |
| 6160 | const writer = buf.writer(); | 5565 | const writer = buf.writer(); |
| 6161 | | 5566 | |
| 6162 | if (self.text_segment_cmd_index) |text_segment_cmd_index| blk: { | 5567 | if (self.stubs_section_index) |sect_id| { |
| 6163 | const stubs_section_index = self.stubs_section_index orelse break :blk; | 5568 | const stubs = &self.sections.items(.header)[sect_id]; |
| 6164 | const stubs = self.getSectionPtr(.{ | | |
| 6165 | .seg = text_segment_cmd_index, | | |
| 6166 | .sect = stubs_section_index, | | |
| 6167 | }); | | |
| 6168 | stubs.reserved1 = 0; | 5569 | stubs.reserved1 = 0; |
| 6169 | for (self.stubs.items) |entry| { | 5570 | for (self.stubs.items) |entry| { |
| 6170 | if (entry.sym_index == 0) continue; | 5571 | if (entry.sym_index == 0) continue; |
| ... | @@ -6172,16 +5573,12 @@ fn writeSymtab(self: *MachO) !void { | ... | @@ -6172,16 +5573,12 @@ fn writeSymtab(self: *MachO) !void { |
| 6172 | if (atom_sym.n_desc == N_DESC_GCED) continue; | 5573 | if (atom_sym.n_desc == N_DESC_GCED) continue; |
| 6173 | const target_sym = self.getSymbol(entry.target); | 5574 | const target_sym = self.getSymbol(entry.target); |
| 6174 | assert(target_sym.undf()); | 5575 | assert(target_sym.undf()); |
| 6175 | try writer.writeIntLittle(u32, dysymtab.iundefsym + imports_table.get(entry.target).?); | 5576 | try writer.writeIntLittle(u32, iundefsym + ctx.imports_table.get(entry.target).?); |
| 6176 | } | 5577 | } |
| 6177 | } | 5578 | } |
| 6178 | | 5579 | |
| 6179 | if (self.data_const_segment_cmd_index) |data_const_segment_cmd_index| blk: { | 5580 | if (self.got_section_index) |sect_id| { |
| 6180 | const got_section_index = self.got_section_index orelse break :blk; | 5581 | const got = &self.sections.items(.header)[sect_id]; |
| 6181 | const got = self.getSectionPtr(.{ | | |
| 6182 | .seg = data_const_segment_cmd_index, | | |
| 6183 | .sect = got_section_index, | | |
| 6184 | }); | | |
| 6185 | got.reserved1 = nstubs; | 5582 | got.reserved1 = nstubs; |
| 6186 | for (self.got_entries.items) |entry| { | 5583 | for (self.got_entries.items) |entry| { |
| 6187 | if (entry.sym_index == 0) continue; | 5584 | if (entry.sym_index == 0) continue; |
| ... | @@ -6189,19 +5586,15 @@ fn writeSymtab(self: *MachO) !void { | ... | @@ -6189,19 +5586,15 @@ fn writeSymtab(self: *MachO) !void { |
| 6189 | if (atom_sym.n_desc == N_DESC_GCED) continue; | 5586 | if (atom_sym.n_desc == N_DESC_GCED) continue; |
| 6190 | const target_sym = self.getSymbol(entry.target); | 5587 | const target_sym = self.getSymbol(entry.target); |
| 6191 | if (target_sym.undf()) { | 5588 | if (target_sym.undf()) { |
| 6192 | try writer.writeIntLittle(u32, dysymtab.iundefsym + imports_table.get(entry.target).?); | 5589 | try writer.writeIntLittle(u32, iundefsym + ctx.imports_table.get(entry.target).?); |
| 6193 | } else { | 5590 | } else { |
| 6194 | try writer.writeIntLittle(u32, macho.INDIRECT_SYMBOL_LOCAL); | 5591 | try writer.writeIntLittle(u32, macho.INDIRECT_SYMBOL_LOCAL); |
| 6195 | } | 5592 | } |
| 6196 | } | 5593 | } |
| 6197 | } | 5594 | } |
| 6198 | | 5595 | |
| 6199 | if (self.data_segment_cmd_index) |data_segment_cmd_index| blk: { | 5596 | if (self.la_symbol_ptr_section_index) |sect_id| { |
| 6200 | const la_symbol_ptr_section_index = self.la_symbol_ptr_section_index orelse break :blk; | 5597 | const la_symbol_ptr = &self.sections.items(.header)[sect_id]; |
| 6201 | const la_symbol_ptr = self.getSectionPtr(.{ | | |
| 6202 | .seg = data_segment_cmd_index, | | |
| 6203 | .sect = la_symbol_ptr_section_index, | | |
| 6204 | }); | | |
| 6205 | la_symbol_ptr.reserved1 = nstubs + ngot_entries; | 5598 | la_symbol_ptr.reserved1 = nstubs + ngot_entries; |
| 6206 | for (self.stubs.items) |entry| { | 5599 | for (self.stubs.items) |entry| { |
| 6207 | if (entry.sym_index == 0) continue; | 5600 | if (entry.sym_index == 0) continue; |
| ... | @@ -6209,131 +5602,76 @@ fn writeSymtab(self: *MachO) !void { | ... | @@ -6209,131 +5602,76 @@ fn writeSymtab(self: *MachO) !void { |
| 6209 | if (atom_sym.n_desc == N_DESC_GCED) continue; | 5602 | if (atom_sym.n_desc == N_DESC_GCED) continue; |
| 6210 | const target_sym = self.getSymbol(entry.target); | 5603 | const target_sym = self.getSymbol(entry.target); |
| 6211 | assert(target_sym.undf()); | 5604 | assert(target_sym.undf()); |
| 6212 | try writer.writeIntLittle(u32, dysymtab.iundefsym + imports_table.get(entry.target).?); | 5605 | try writer.writeIntLittle(u32, iundefsym + ctx.imports_table.get(entry.target).?); |
| 6213 | } | 5606 | } |
| 6214 | } | 5607 | } |
| 6215 | | 5608 | |
| 6216 | assert(buf.items.len == dysymtab.nindirectsyms * @sizeOf(u32)); | 5609 | assert(buf.items.len == needed_size); |
| 6217 | | 5610 | try self.base.file.?.pwriteAll(buf.items, offset); |
| 6218 | try self.base.file.?.pwriteAll(buf.items, dysymtab.indirectsymoff); | | |
| 6219 | self.load_commands_dirty = true; | | |
| 6220 | } | | |
| 6221 | | | |
| 6222 | fn writeStrtab(self: *MachO) !void { | | |
| 6223 | const tracy = trace(@src()); | | |
| 6224 | defer tracy.end(); | | |
| 6225 | | | |
| 6226 | const seg = &self.load_commands.items[self.linkedit_segment_cmd_index.?].segment; | | |
| 6227 | const symtab = &self.load_commands.items[self.symtab_cmd_index.?].symtab; | | |
| 6228 | const stroff = mem.alignForwardGeneric(u64, seg.inner.fileoff + seg.inner.filesize, @alignOf(u64)); | | |
| 6229 | | | |
| 6230 | const strsize = self.strtab.buffer.items.len; | | |
| 6231 | symtab.stroff = @intCast(u32, stroff); | | |
| 6232 | symtab.strsize = @intCast(u32, strsize); | | |
| 6233 | seg.inner.filesize = symtab.stroff + symtab.strsize - seg.inner.fileoff; | | |
| 6234 | | | |
| 6235 | log.debug("writing string table from 0x{x} to 0x{x}", .{ symtab.stroff, symtab.stroff + symtab.strsize }); | | |
| 6236 | | | |
| 6237 | try self.base.file.?.pwriteAll(self.strtab.buffer.items, symtab.stroff); | | |
| 6238 | | | |
| 6239 | self.load_commands_dirty = true; | | |
| 6240 | } | | |
| 6241 | | | |
| 6242 | fn writeLinkeditSegment(self: *MachO) !void { | | |
| 6243 | const tracy = trace(@src()); | | |
| 6244 | defer tracy.end(); | | |
| 6245 | | 5611 | |
| 6246 | const seg = &self.load_commands.items[self.linkedit_segment_cmd_index.?].segment; | 5612 | lc.nlocalsym = ctx.nlocalsym; |
| 6247 | seg.inner.filesize = 0; | 5613 | lc.iextdefsym = iextdefsym; |
| 6248 | | 5614 | lc.nextdefsym = ctx.nextdefsym; |
| 6249 | try self.writeDyldInfoData(); | 5615 | lc.iundefsym = iundefsym; |
| 6250 | try self.writeFunctionStarts(); | 5616 | lc.nundefsym = ctx.nundefsym; |
| 6251 | try self.writeDataInCode(); | 5617 | lc.indirectsymoff = @intCast(u32, offset); |
| 6252 | try self.writeSymtab(); | 5618 | lc.nindirectsyms = nindirectsyms; |
| 6253 | try self.writeStrtab(); | | |
| 6254 | | | |
| 6255 | seg.inner.vmsize = mem.alignForwardGeneric(u64, seg.inner.filesize, self.page_size); | | |
| 6256 | } | 5619 | } |
| 6257 | | 5620 | |
| 6258 | fn writeCodeSignaturePadding(self: *MachO, code_sig: *CodeSignature) !void { | 5621 | fn writeCodeSignaturePadding( |
| 6259 | const tracy = trace(@src()); | 5622 | self: *MachO, |
| 6260 | defer tracy.end(); | 5623 | code_sig: *CodeSignature, |
| 6261 | | 5624 | ncmds: *u32, |
| 6262 | const seg = &self.load_commands.items[self.linkedit_segment_cmd_index.?].segment; | 5625 | lc_writer: anytype, |
| 6263 | const cs_cmd = &self.load_commands.items[self.code_signature_cmd_index.?].linkedit_data; | 5626 | ) !u32 { |
| | 5627 | const seg = &self.segments.items[self.linkedit_segment_cmd_index.?]; |
| 6264 | // Code signature data has to be 16-bytes aligned for Apple tools to recognize the file | 5628 | // Code signature data has to be 16-bytes aligned for Apple tools to recognize the file |
| 6265 | // https://github.com/opensource-apple/cctools/blob/fdb4825f303fd5c0751be524babd32958181b3ed/libstuff/checkout.c#L271 | 5629 | // https://github.com/opensource-apple/cctools/blob/fdb4825f303fd5c0751be524babd32958181b3ed/libstuff/checkout.c#L271 |
| 6266 | const dataoff = mem.alignForwardGeneric(u64, seg.inner.fileoff + seg.inner.filesize, 16); | 5630 | const offset = mem.alignForwardGeneric(u64, seg.fileoff + seg.filesize, 16); |
| 6267 | const datasize = code_sig.estimateSize(dataoff); | 5631 | const needed_size = code_sig.estimateSize(offset); |
| 6268 | cs_cmd.dataoff = @intCast(u32, dataoff); | 5632 | seg.filesize = offset + needed_size - seg.fileoff; |
| 6269 | cs_cmd.datasize = @intCast(u32, code_sig.estimateSize(dataoff)); | 5633 | seg.vmsize = mem.alignForwardGeneric(u64, seg.filesize, self.page_size); |
| 6270 | | 5634 | log.debug("writing code signature padding from 0x{x} to 0x{x}", .{ offset, offset + needed_size }); |
| 6271 | // Advance size of __LINKEDIT segment | | |
| 6272 | seg.inner.filesize = cs_cmd.dataoff + cs_cmd.datasize - seg.inner.fileoff; | | |
| 6273 | seg.inner.vmsize = mem.alignForwardGeneric(u64, seg.inner.filesize, self.page_size); | | |
| 6274 | log.debug("writing code signature padding from 0x{x} to 0x{x}", .{ dataoff, dataoff + datasize }); | | |
| 6275 | // Pad out the space. We need to do this to calculate valid hashes for everything in the file | 5635 | // Pad out the space. We need to do this to calculate valid hashes for everything in the file |
| 6276 | // except for code signature data. | 5636 | // except for code signature data. |
| 6277 | try self.base.file.?.pwriteAll(&[_]u8{0}, dataoff + datasize - 1); | 5637 | try self.base.file.?.pwriteAll(&[_]u8{0}, offset + needed_size - 1); |
| 6278 | self.load_commands_dirty = true; | | |
| 6279 | } | | |
| 6280 | | 5638 | |
| 6281 | fn writeCodeSignature(self: *MachO, code_sig: *CodeSignature) !void { | 5639 | try lc_writer.writeStruct(macho.linkedit_data_command{ |
| 6282 | const tracy = trace(@src()); | 5640 | .cmd = .CODE_SIGNATURE, |
| 6283 | defer tracy.end(); | 5641 | .cmdsize = @sizeOf(macho.linkedit_data_command), |
| | 5642 | .dataoff = @intCast(u32, offset), |
| | 5643 | .datasize = @intCast(u32, needed_size), |
| | 5644 | }); |
| | 5645 | ncmds.* += 1; |
| | 5646 | |
| | 5647 | return @intCast(u32, offset); |
| | 5648 | } |
| 6284 | | 5649 | |
| 6285 | const code_sig_cmd = self.load_commands.items[self.code_signature_cmd_index.?].linkedit_data; | 5650 | fn writeCodeSignature(self: *MachO, code_sig: *CodeSignature, offset: u32) !void { |
| 6286 | const seg = self.load_commands.items[self.text_segment_cmd_index.?].segment; | 5651 | const seg = self.segments.items[self.text_segment_cmd_index.?]; |
| 6287 | | 5652 | |
| 6288 | var buffer = std.ArrayList(u8).init(self.base.allocator); | 5653 | var buffer = std.ArrayList(u8).init(self.base.allocator); |
| 6289 | defer buffer.deinit(); | 5654 | defer buffer.deinit(); |
| 6290 | try buffer.ensureTotalCapacityPrecise(code_sig.size()); | 5655 | try buffer.ensureTotalCapacityPrecise(code_sig.size()); |
| 6291 | try code_sig.writeAdhocSignature(self.base.allocator, .{ | 5656 | try code_sig.writeAdhocSignature(self.base.allocator, .{ |
| 6292 | .file = self.base.file.?, | 5657 | .file = self.base.file.?, |
| 6293 | .exec_seg_base = seg.inner.fileoff, | 5658 | .exec_seg_base = seg.fileoff, |
| 6294 | .exec_seg_limit = seg.inner.filesize, | 5659 | .exec_seg_limit = seg.filesize, |
| 6295 | .code_sig_cmd = code_sig_cmd, | 5660 | .file_size = offset, |
| 6296 | .output_mode = self.base.options.output_mode, | 5661 | .output_mode = self.base.options.output_mode, |
| 6297 | }, buffer.writer()); | 5662 | }, buffer.writer()); |
| 6298 | assert(buffer.items.len == code_sig.size()); | 5663 | assert(buffer.items.len == code_sig.size()); |
| 6299 | | 5664 | |
| 6300 | log.debug("writing code signature from 0x{x} to 0x{x}", .{ | 5665 | log.debug("writing code signature from 0x{x} to 0x{x}", .{ |
| 6301 | code_sig_cmd.dataoff, | 5666 | offset, |
| 6302 | code_sig_cmd.dataoff + buffer.items.len, | 5667 | offset + buffer.items.len, |
| 6303 | }); | 5668 | }); |
| 6304 | | 5669 | |
| 6305 | try self.base.file.?.pwriteAll(buffer.items, code_sig_cmd.dataoff); | 5670 | try self.base.file.?.pwriteAll(buffer.items, offset); |
| 6306 | } | | |
| 6307 | | | |
| 6308 | /// Writes all load commands and section headers. | | |
| 6309 | fn writeLoadCommands(self: *MachO) !void { | | |
| 6310 | if (!self.load_commands_dirty) return; | | |
| 6311 | | | |
| 6312 | var sizeofcmds: u32 = 0; | | |
| 6313 | for (self.load_commands.items) |lc| { | | |
| 6314 | if (lc.cmd() == .NONE) continue; | | |
| 6315 | sizeofcmds += lc.cmdsize(); | | |
| 6316 | } | | |
| 6317 | | | |
| 6318 | var buffer = try self.base.allocator.alloc(u8, sizeofcmds); | | |
| 6319 | defer self.base.allocator.free(buffer); | | |
| 6320 | var fib = std.io.fixedBufferStream(buffer); | | |
| 6321 | const writer = fib.writer(); | | |
| 6322 | for (self.load_commands.items) |lc| { | | |
| 6323 | if (lc.cmd() == .NONE) continue; | | |
| 6324 | try lc.write(writer); | | |
| 6325 | } | | |
| 6326 | | | |
| 6327 | const off = @sizeOf(macho.mach_header_64); | | |
| 6328 | | | |
| 6329 | log.debug("writing load commands from 0x{x} to 0x{x}", .{ off, off + sizeofcmds }); | | |
| 6330 | | | |
| 6331 | try self.base.file.?.pwriteAll(buffer, off); | | |
| 6332 | self.load_commands_dirty = false; | | |
| 6333 | } | 5671 | } |
| 6334 | | 5672 | |
| 6335 | /// Writes Mach-O file header. | 5673 | /// Writes Mach-O file header. |
| 6336 | fn writeHeader(self: *MachO) !void { | 5674 | fn writeHeader(self: *MachO, ncmds: u32, sizeofcmds: u32) !void { |
| 6337 | var header: macho.mach_header_64 = .{}; | 5675 | var header: macho.mach_header_64 = .{}; |
| 6338 | header.flags = macho.MH_NOUNDEFS | macho.MH_DYLDLINK | macho.MH_PIE | macho.MH_TWOLEVEL; | 5676 | header.flags = macho.MH_NOUNDEFS | macho.MH_DYLDLINK | macho.MH_PIE | macho.MH_TWOLEVEL; |
| 6339 | | 5677 | |
| ... | @@ -6361,18 +5699,12 @@ fn writeHeader(self: *MachO) !void { | ... | @@ -6361,18 +5699,12 @@ fn writeHeader(self: *MachO) !void { |
| 6361 | else => unreachable, | 5699 | else => unreachable, |
| 6362 | } | 5700 | } |
| 6363 | | 5701 | |
| 6364 | if (self.tlv_section_index) |_| { | 5702 | if (self.getSectionByName("__DATA", "__thread_vars")) |_| { |
| 6365 | header.flags |= macho.MH_HAS_TLV_DESCRIPTORS; | 5703 | header.flags |= macho.MH_HAS_TLV_DESCRIPTORS; |
| 6366 | } | 5704 | } |
| 6367 | | 5705 | |
| 6368 | header.ncmds = 0; | 5706 | header.ncmds = ncmds; |
| 6369 | header.sizeofcmds = 0; | 5707 | header.sizeofcmds = sizeofcmds; |
| 6370 | | | |
| 6371 | for (self.load_commands.items) |cmd| { | | |
| 6372 | if (cmd.cmd() == .NONE) continue; | | |
| 6373 | header.sizeofcmds += cmd.cmdsize(); | | |
| 6374 | header.ncmds += 1; | | |
| 6375 | } | | |
| 6376 | | 5708 | |
| 6377 | log.debug("writing Mach-O header {}", .{header}); | 5709 | log.debug("writing Mach-O header {}", .{header}); |
| 6378 | | 5710 | |
| ... | @@ -6392,33 +5724,27 @@ pub fn makeStaticString(bytes: []const u8) [16]u8 { | ... | @@ -6392,33 +5724,27 @@ pub fn makeStaticString(bytes: []const u8) [16]u8 { |
| 6392 | return buf; | 5724 | return buf; |
| 6393 | } | 5725 | } |
| 6394 | | 5726 | |
| 6395 | pub fn getSectionOrdinal(self: *MachO, match: MatchingSection) u8 { | 5727 | fn getSegmentByName(self: MachO, segname: []const u8) ?u8 { |
| 6396 | return @intCast(u8, self.section_ordinals.getIndex(match).?) + 1; | 5728 | for (self.segments.items) |seg, i| { |
| 6397 | } | 5729 | if (mem.eql(u8, segname, seg.segName())) return @intCast(u8, i); |
| 6398 | | 5730 | } else return null; |
| 6399 | pub fn getMatchingSectionFromOrdinal(self: *MachO, ord: u8) MatchingSection { | | |
| 6400 | const index = ord - 1; | | |
| 6401 | assert(index < self.section_ordinals.count()); | | |
| 6402 | return self.section_ordinals.keys()[index]; | | |
| 6403 | } | | |
| 6404 | | | |
| 6405 | pub fn getSegmentPtr(self: *MachO, match: MatchingSection) *macho.SegmentCommand { | | |
| 6406 | assert(match.seg < self.load_commands.items.len); | | |
| 6407 | return &self.load_commands.items[match.seg].segment; | | |
| 6408 | } | | |
| 6409 | | | |
| 6410 | pub fn getSegment(self: *MachO, match: MatchingSection) macho.SegmentCommand { | | |
| 6411 | return self.getSegmentPtr(match).*; | | |
| 6412 | } | 5731 | } |
| 6413 | | 5732 | |
| 6414 | pub fn getSectionPtr(self: *MachO, match: MatchingSection) *macho.section_64 { | 5733 | pub fn getSectionByName(self: MachO, segname: []const u8, sectname: []const u8) ?u8 { |
| 6415 | const seg = self.getSegmentPtr(match); | 5734 | // TODO investigate caching with a hashmap |
| 6416 | assert(match.sect < seg.sections.items.len); | 5735 | for (self.sections.items(.header)) |header, i| { |
| 6417 | return &seg.sections.items[match.sect]; | 5736 | if (mem.eql(u8, header.segName(), segname) and mem.eql(u8, header.sectName(), sectname)) |
| | 5737 | return @intCast(u8, i); |
| | 5738 | } else return null; |
| 6418 | } | 5739 | } |
| 6419 | | 5740 | |
| 6420 | pub fn getSection(self: *MachO, match: MatchingSection) macho.section_64 { | 5741 | pub fn getSectionIndexes(self: MachO, segment_index: u8) struct { start: u8, end: u8 } { |
| 6421 | return self.getSectionPtr(match).*; | 5742 | var start: u8 = 0; |
| | 5743 | const nsects = for (self.segments.items) |seg, i| { |
| | 5744 | if (i == segment_index) break @intCast(u8, seg.nsects); |
| | 5745 | start += @intCast(u8, seg.nsects); |
| | 5746 | } else 0; |
| | 5747 | return .{ .start = start, .end = start + nsects }; |
| 6422 | } | 5748 | } |
| 6423 | | 5749 | |
| 6424 | pub fn symbolIsTemp(self: *MachO, sym_with_loc: SymbolWithLoc) bool { | 5750 | pub fn symbolIsTemp(self: *MachO, sym_with_loc: SymbolWithLoc) bool { |
| ... | @@ -6512,72 +5838,6 @@ pub fn findFirst(comptime T: type, haystack: []const T, start: usize, predicate: | ... | @@ -6512,72 +5838,6 @@ pub fn findFirst(comptime T: type, haystack: []const T, start: usize, predicate: |
| 6512 | return i; | 5838 | return i; |
| 6513 | } | 5839 | } |
| 6514 | | 5840 | |
| 6515 | const DebugInfo = struct { | | |
| 6516 | inner: dwarf.DwarfInfo, | | |
| 6517 | debug_info: []const u8, | | |
| 6518 | debug_abbrev: []const u8, | | |
| 6519 | debug_str: []const u8, | | |
| 6520 | debug_line: []const u8, | | |
| 6521 | debug_line_str: []const u8, | | |
| 6522 | debug_ranges: []const u8, | | |
| 6523 | | | |
| 6524 | pub fn parse(allocator: Allocator, object: Object) !?DebugInfo { | | |
| 6525 | var debug_info = blk: { | | |
| 6526 | const index = object.dwarf_debug_info_index orelse return null; | | |
| 6527 | break :blk try object.getSectionContents(index); | | |
| 6528 | }; | | |
| 6529 | var debug_abbrev = blk: { | | |
| 6530 | const index = object.dwarf_debug_abbrev_index orelse return null; | | |
| 6531 | break :blk try object.getSectionContents(index); | | |
| 6532 | }; | | |
| 6533 | var debug_str = blk: { | | |
| 6534 | const index = object.dwarf_debug_str_index orelse return null; | | |
| 6535 | break :blk try object.getSectionContents(index); | | |
| 6536 | }; | | |
| 6537 | var debug_line = blk: { | | |
| 6538 | const index = object.dwarf_debug_line_index orelse return null; | | |
| 6539 | break :blk try object.getSectionContents(index); | | |
| 6540 | }; | | |
| 6541 | var debug_line_str = blk: { | | |
| 6542 | if (object.dwarf_debug_line_str_index) |ind| { | | |
| 6543 | break :blk try object.getSectionContents(ind); | | |
| 6544 | } | | |
| 6545 | break :blk &[0]u8{}; | | |
| 6546 | }; | | |
| 6547 | var debug_ranges = blk: { | | |
| 6548 | if (object.dwarf_debug_ranges_index) |ind| { | | |
| 6549 | break :blk try object.getSectionContents(ind); | | |
| 6550 | } | | |
| 6551 | break :blk &[0]u8{}; | | |
| 6552 | }; | | |
| 6553 | | | |
| 6554 | var inner: dwarf.DwarfInfo = .{ | | |
| 6555 | .endian = .Little, | | |
| 6556 | .debug_info = debug_info, | | |
| 6557 | .debug_abbrev = debug_abbrev, | | |
| 6558 | .debug_str = debug_str, | | |
| 6559 | .debug_line = debug_line, | | |
| 6560 | .debug_line_str = debug_line_str, | | |
| 6561 | .debug_ranges = debug_ranges, | | |
| 6562 | }; | | |
| 6563 | try dwarf.openDwarfDebugInfo(&inner, allocator); | | |
| 6564 | | | |
| 6565 | return DebugInfo{ | | |
| 6566 | .inner = inner, | | |
| 6567 | .debug_info = debug_info, | | |
| 6568 | .debug_abbrev = debug_abbrev, | | |
| 6569 | .debug_str = debug_str, | | |
| 6570 | .debug_line = debug_line, | | |
| 6571 | .debug_line_str = debug_line_str, | | |
| 6572 | .debug_ranges = debug_ranges, | | |
| 6573 | }; | | |
| 6574 | } | | |
| 6575 | | | |
| 6576 | pub fn deinit(self: *DebugInfo, allocator: Allocator) void { | | |
| 6577 | self.inner.deinit(allocator); | | |
| 6578 | } | | |
| 6579 | }; | | |
| 6580 | | | |
| 6581 | pub fn generateSymbolStabs( | 5841 | pub fn generateSymbolStabs( |
| 6582 | self: *MachO, | 5842 | self: *MachO, |
| 6583 | object: Object, | 5843 | object: Object, |
| ... | @@ -6585,14 +5845,15 @@ pub fn generateSymbolStabs( | ... | @@ -6585,14 +5845,15 @@ pub fn generateSymbolStabs( |
| 6585 | ) !void { | 5845 | ) !void { |
| 6586 | assert(!self.base.options.strip); | 5846 | assert(!self.base.options.strip); |
| 6587 | | 5847 | |
| 6588 | const gpa = self.base.allocator; | | |
| 6589 | | | |
| 6590 | log.debug("parsing debug info in '{s}'", .{object.name}); | 5848 | log.debug("parsing debug info in '{s}'", .{object.name}); |
| 6591 | | 5849 | |
| 6592 | var debug_info = (try DebugInfo.parse(gpa, object)) orelse return; | 5850 | const gpa = self.base.allocator; |
| | 5851 | var debug_info = try object.parseDwarfInfo(); |
| | 5852 | defer debug_info.deinit(gpa); |
| | 5853 | try dwarf.openDwarfDebugInfo(&debug_info, gpa); |
| 6593 | | 5854 | |
| 6594 | // We assume there is only one CU. | 5855 | // We assume there is only one CU. |
| 6595 | const compile_unit = debug_info.inner.findCompileUnit(0x0) catch |err| switch (err) { | 5856 | const compile_unit = debug_info.findCompileUnit(0x0) catch |err| switch (err) { |
| 6596 | error.MissingDebugInfo => { | 5857 | error.MissingDebugInfo => { |
| 6597 | // TODO audit cases with missing debug info and audit our dwarf.zig module. | 5858 | // TODO audit cases with missing debug info and audit our dwarf.zig module. |
| 6598 | log.debug("invalid or missing debug info in {s}; skipping", .{object.name}); | 5859 | log.debug("invalid or missing debug info in {s}; skipping", .{object.name}); |
| ... | @@ -6600,8 +5861,8 @@ pub fn generateSymbolStabs( | ... | @@ -6600,8 +5861,8 @@ pub fn generateSymbolStabs( |
| 6600 | }, | 5861 | }, |
| 6601 | else => |e| return e, | 5862 | else => |e| return e, |
| 6602 | }; | 5863 | }; |
| 6603 | const tu_name = try compile_unit.die.getAttrString(&debug_info.inner, dwarf.AT.name); | 5864 | const tu_name = try compile_unit.die.getAttrString(&debug_info, dwarf.AT.name); |
| 6604 | const tu_comp_dir = try compile_unit.die.getAttrString(&debug_info.inner, dwarf.AT.comp_dir); | 5865 | const tu_comp_dir = try compile_unit.die.getAttrString(&debug_info, dwarf.AT.comp_dir); |
| 6605 | | 5866 | |
| 6606 | // Open scope | 5867 | // Open scope |
| 6607 | try locals.ensureUnusedCapacity(3); | 5868 | try locals.ensureUnusedCapacity(3); |
| ... | @@ -6664,7 +5925,7 @@ pub fn generateSymbolStabs( | ... | @@ -6664,7 +5925,7 @@ pub fn generateSymbolStabs( |
| 6664 | fn generateSymbolStabsForSymbol( | 5925 | fn generateSymbolStabsForSymbol( |
| 6665 | self: *MachO, | 5926 | self: *MachO, |
| 6666 | sym_loc: SymbolWithLoc, | 5927 | sym_loc: SymbolWithLoc, |
| 6667 | debug_info: DebugInfo, | 5928 | debug_info: dwarf.DwarfInfo, |
| 6668 | buf: *[4]macho.nlist_64, | 5929 | buf: *[4]macho.nlist_64, |
| 6669 | ) ![]const macho.nlist_64 { | 5930 | ) ![]const macho.nlist_64 { |
| 6670 | const gpa = self.base.allocator; | 5931 | const gpa = self.base.allocator; |
| ... | @@ -6679,7 +5940,7 @@ fn generateSymbolStabsForSymbol( | ... | @@ -6679,7 +5940,7 @@ fn generateSymbolStabsForSymbol( |
| 6679 | const source_sym = object.getSourceSymbol(sym_loc.sym_index) orelse return buf[0..0]; | 5940 | const source_sym = object.getSourceSymbol(sym_loc.sym_index) orelse return buf[0..0]; |
| 6680 | const size: ?u64 = size: { | 5941 | const size: ?u64 = size: { |
| 6681 | if (source_sym.tentative()) break :size null; | 5942 | if (source_sym.tentative()) break :size null; |
| 6682 | for (debug_info.inner.func_list.items) |func| { | 5943 | for (debug_info.func_list.items) |func| { |
| 6683 | if (func.pc_range) |range| { | 5944 | if (func.pc_range) |range| { |
| 6684 | if (source_sym.n_value >= range.start and source_sym.n_value < range.end) { | 5945 | if (source_sym.n_value >= range.start and source_sym.n_value < range.end) { |
| 6685 | break :size range.end - range.start; | 5946 | break :size range.end - range.start; |
| ... | @@ -6731,259 +5992,272 @@ fn generateSymbolStabsForSymbol( | ... | @@ -6731,259 +5992,272 @@ fn generateSymbolStabsForSymbol( |
| 6731 | } | 5992 | } |
| 6732 | } | 5993 | } |
| 6733 | | 5994 | |
| 6734 | fn snapshotState(self: *MachO) !void { | 5995 | // fn snapshotState(self: *MachO) !void { |
| 6735 | const emit = self.base.options.emit orelse { | 5996 | // const emit = self.base.options.emit orelse { |
| 6736 | log.debug("no emit directory found; skipping snapshot...", .{}); | 5997 | // log.debug("no emit directory found; skipping snapshot...", .{}); |
| 6737 | return; | 5998 | // return; |
| 6738 | }; | 5999 | // }; |
| 6739 | | 6000 | |
| 6740 | const Snapshot = struct { | 6001 | // const Snapshot = struct { |
| 6741 | const Node = struct { | 6002 | // const Node = struct { |
| 6742 | const Tag = enum { | 6003 | // const Tag = enum { |
| 6743 | section_start, | 6004 | // section_start, |
| 6744 | section_end, | 6005 | // section_end, |
| 6745 | atom_start, | 6006 | // atom_start, |
| 6746 | atom_end, | 6007 | // atom_end, |
| 6747 | relocation, | 6008 | // relocation, |
| 6748 | | 6009 | |
| 6749 | pub fn jsonStringify( | 6010 | // pub fn jsonStringify( |
| 6750 | tag: Tag, | 6011 | // tag: Tag, |
| 6751 | options: std.json.StringifyOptions, | 6012 | // options: std.json.StringifyOptions, |
| 6752 | out_stream: anytype, | 6013 | // out_stream: anytype, |
| 6753 | ) !void { | 6014 | // ) !void { |
| 6754 | _ = options; | 6015 | // _ = options; |
| 6755 | switch (tag) { | 6016 | // switch (tag) { |
| 6756 | .section_start => try out_stream.writeAll("\"section_start\""), | 6017 | // .section_start => try out_stream.writeAll("\"section_start\""), |
| 6757 | .section_end => try out_stream.writeAll("\"section_end\""), | 6018 | // .section_end => try out_stream.writeAll("\"section_end\""), |
| 6758 | .atom_start => try out_stream.writeAll("\"atom_start\""), | 6019 | // .atom_start => try out_stream.writeAll("\"atom_start\""), |
| 6759 | .atom_end => try out_stream.writeAll("\"atom_end\""), | 6020 | // .atom_end => try out_stream.writeAll("\"atom_end\""), |
| 6760 | .relocation => try out_stream.writeAll("\"relocation\""), | 6021 | // .relocation => try out_stream.writeAll("\"relocation\""), |
| 6761 | } | 6022 | // } |
| 6762 | } | 6023 | // } |
| 6763 | }; | 6024 | // }; |
| 6764 | const Payload = struct { | 6025 | // const Payload = struct { |
| 6765 | name: []const u8 = "", | 6026 | // name: []const u8 = "", |
| 6766 | aliases: [][]const u8 = &[0][]const u8{}, | 6027 | // aliases: [][]const u8 = &[0][]const u8{}, |
| 6767 | is_global: bool = false, | 6028 | // is_global: bool = false, |
| 6768 | target: u64 = 0, | 6029 | // target: u64 = 0, |
| 6769 | }; | 6030 | // }; |
| 6770 | address: u64, | 6031 | // address: u64, |
| 6771 | tag: Tag, | 6032 | // tag: Tag, |
| 6772 | payload: Payload, | 6033 | // payload: Payload, |
| 6773 | }; | 6034 | // }; |
| 6774 | timestamp: i128, | 6035 | // timestamp: i128, |
| 6775 | nodes: []Node, | 6036 | // nodes: []Node, |
| 6776 | }; | 6037 | // }; |
| 6777 | | 6038 | |
| 6778 | var arena_allocator = std.heap.ArenaAllocator.init(self.base.allocator); | 6039 | // var arena_allocator = std.heap.ArenaAllocator.init(self.base.allocator); |
| 6779 | defer arena_allocator.deinit(); | 6040 | // defer arena_allocator.deinit(); |
| 6780 | const arena = arena_allocator.allocator(); | 6041 | // const arena = arena_allocator.allocator(); |
| 6781 | | 6042 | |
| 6782 | const out_file = try emit.directory.handle.createFile("snapshots.json", .{ | 6043 | // const out_file = try emit.directory.handle.createFile("snapshots.json", .{ |
| 6783 | .truncate = false, | 6044 | // .truncate = false, |
| 6784 | .read = true, | 6045 | // .read = true, |
| 6785 | }); | 6046 | // }); |
| 6786 | defer out_file.close(); | 6047 | // defer out_file.close(); |
| 6787 | | 6048 | |
| 6788 | if (out_file.seekFromEnd(-1)) { | 6049 | // if (out_file.seekFromEnd(-1)) { |
| 6789 | try out_file.writer().writeByte(','); | 6050 | // try out_file.writer().writeByte(','); |
| 6790 | } else |err| switch (err) { | 6051 | // } else |err| switch (err) { |
| 6791 | error.Unseekable => try out_file.writer().writeByte('['), | 6052 | // error.Unseekable => try out_file.writer().writeByte('['), |
| 6792 | else => |e| return e, | 6053 | // else => |e| return e, |
| 6793 | } | 6054 | // } |
| 6794 | const writer = out_file.writer(); | 6055 | // const writer = out_file.writer(); |
| 6795 | | 6056 | |
| 6796 | var snapshot = Snapshot{ | 6057 | // var snapshot = Snapshot{ |
| 6797 | .timestamp = std.time.nanoTimestamp(), | 6058 | // .timestamp = std.time.nanoTimestamp(), |
| 6798 | .nodes = undefined, | 6059 | // .nodes = undefined, |
| 6799 | }; | 6060 | // }; |
| 6800 | var nodes = std.ArrayList(Snapshot.Node).init(arena); | 6061 | // var nodes = std.ArrayList(Snapshot.Node).init(arena); |
| 6801 | | 6062 | |
| 6802 | for (self.section_ordinals.keys()) |key| { | 6063 | // for (self.section_ordinals.keys()) |key| { |
| 6803 | const sect = self.getSection(key); | 6064 | // const sect = self.getSection(key); |
| 6804 | const sect_name = try std.fmt.allocPrint(arena, "{s},{s}", .{ sect.segName(), sect.sectName() }); | 6065 | // const sect_name = try std.fmt.allocPrint(arena, "{s},{s}", .{ sect.segName(), sect.sectName() }); |
| 6805 | try nodes.append(.{ | 6066 | // try nodes.append(.{ |
| 6806 | .address = sect.addr, | 6067 | // .address = sect.addr, |
| 6807 | .tag = .section_start, | 6068 | // .tag = .section_start, |
| 6808 | .payload = .{ .name = sect_name }, | 6069 | // .payload = .{ .name = sect_name }, |
| 6809 | }); | 6070 | // }); |
| 6810 | | 6071 | |
| 6811 | const is_tlv = sect.type_() == macho.S_THREAD_LOCAL_VARIABLES; | 6072 | // const is_tlv = sect.type_() == macho.S_THREAD_LOCAL_VARIABLES; |
| 6812 | | 6073 | |
| 6813 | var atom: *Atom = self.atoms.get(key) orelse { | 6074 | // var atom: *Atom = self.atoms.get(key) orelse { |
| 6814 | try nodes.append(.{ | 6075 | // try nodes.append(.{ |
| 6815 | .address = sect.addr + sect.size, | 6076 | // .address = sect.addr + sect.size, |
| 6816 | .tag = .section_end, | 6077 | // .tag = .section_end, |
| 6817 | .payload = .{}, | 6078 | // .payload = .{}, |
| 6818 | }); | 6079 | // }); |
| 6819 | continue; | 6080 | // continue; |
| 6820 | }; | 6081 | // }; |
| 6821 | | 6082 | |
| 6822 | while (atom.prev) |prev| { | 6083 | // while (atom.prev) |prev| { |
| 6823 | atom = prev; | 6084 | // atom = prev; |
| 6824 | } | 6085 | // } |
| 6825 | | 6086 | |
| 6826 | while (true) { | 6087 | // while (true) { |
| 6827 | const atom_sym = atom.getSymbol(self); | 6088 | // const atom_sym = atom.getSymbol(self); |
| 6828 | var node = Snapshot.Node{ | 6089 | // var node = Snapshot.Node{ |
| 6829 | .address = atom_sym.n_value, | 6090 | // .address = atom_sym.n_value, |
| 6830 | .tag = .atom_start, | 6091 | // .tag = .atom_start, |
| 6831 | .payload = .{ | 6092 | // .payload = .{ |
| 6832 | .name = atom.getName(self), | 6093 | // .name = atom.getName(self), |
| 6833 | .is_global = self.globals.contains(atom.getName(self)), | 6094 | // .is_global = self.globals.contains(atom.getName(self)), |
| 6834 | }, | 6095 | // }, |
| 6835 | }; | 6096 | // }; |
| 6836 | | 6097 | |
| 6837 | var aliases = std.ArrayList([]const u8).init(arena); | 6098 | // var aliases = std.ArrayList([]const u8).init(arena); |
| 6838 | for (atom.contained.items) |sym_off| { | 6099 | // for (atom.contained.items) |sym_off| { |
| 6839 | if (sym_off.offset == 0) { | 6100 | // if (sym_off.offset == 0) { |
| 6840 | try aliases.append(self.getSymbolName(.{ | 6101 | // try aliases.append(self.getSymbolName(.{ |
| 6841 | .sym_index = sym_off.sym_index, | 6102 | // .sym_index = sym_off.sym_index, |
| 6842 | .file = atom.file, | 6103 | // .file = atom.file, |
| 6843 | })); | 6104 | // })); |
| 6844 | } | 6105 | // } |
| 6845 | } | 6106 | // } |
| 6846 | node.payload.aliases = aliases.toOwnedSlice(); | 6107 | // node.payload.aliases = aliases.toOwnedSlice(); |
| 6847 | try nodes.append(node); | 6108 | // try nodes.append(node); |
| 6848 | | 6109 | |
| 6849 | var relocs = try std.ArrayList(Snapshot.Node).initCapacity(arena, atom.relocs.items.len); | 6110 | // var relocs = try std.ArrayList(Snapshot.Node).initCapacity(arena, atom.relocs.items.len); |
| 6850 | for (atom.relocs.items) |rel| { | 6111 | // for (atom.relocs.items) |rel| { |
| 6851 | const source_addr = blk: { | 6112 | // const source_addr = blk: { |
| 6852 | const source_sym = atom.getSymbol(self); | 6113 | // const source_sym = atom.getSymbol(self); |
| 6853 | break :blk source_sym.n_value + rel.offset; | 6114 | // break :blk source_sym.n_value + rel.offset; |
| 6854 | }; | 6115 | // }; |
| 6855 | const target_addr = blk: { | 6116 | // const target_addr = blk: { |
| 6856 | const target_atom = rel.getTargetAtom(self) orelse { | 6117 | // const target_atom = rel.getTargetAtom(self) orelse { |
| 6857 | // If there is no atom for target, we still need to check for special, atom-less | 6118 | // // If there is no atom for target, we still need to check for special, atom-less |
| 6858 | // symbols such as `___dso_handle`. | 6119 | // // symbols such as `___dso_handle`. |
| 6859 | const target_name = self.getSymbolName(rel.target); | 6120 | // const target_name = self.getSymbolName(rel.target); |
| 6860 | if (self.globals.contains(target_name)) { | 6121 | // if (self.globals.contains(target_name)) { |
| 6861 | const atomless_sym = self.getSymbol(rel.target); | 6122 | // const atomless_sym = self.getSymbol(rel.target); |
| 6862 | break :blk atomless_sym.n_value; | 6123 | // break :blk atomless_sym.n_value; |
| 6863 | } | 6124 | // } |
| 6864 | break :blk 0; | 6125 | // break :blk 0; |
| 6865 | }; | 6126 | // }; |
| 6866 | const target_sym = if (target_atom.isSymbolContained(rel.target, self)) | 6127 | // const target_sym = if (target_atom.isSymbolContained(rel.target, self)) |
| 6867 | self.getSymbol(rel.target) | 6128 | // self.getSymbol(rel.target) |
| 6868 | else | 6129 | // else |
| 6869 | target_atom.getSymbol(self); | 6130 | // target_atom.getSymbol(self); |
| 6870 | const base_address: u64 = if (is_tlv) base_address: { | 6131 | // const base_address: u64 = if (is_tlv) base_address: { |
| 6871 | const sect_id: u16 = sect_id: { | 6132 | // const sect_id: u16 = sect_id: { |
| 6872 | if (self.tlv_data_section_index) |i| { | 6133 | // if (self.tlv_data_section_index) |i| { |
| 6873 | break :sect_id i; | 6134 | // break :sect_id i; |
| 6874 | } else if (self.tlv_bss_section_index) |i| { | 6135 | // } else if (self.tlv_bss_section_index) |i| { |
| 6875 | break :sect_id i; | 6136 | // break :sect_id i; |
| 6876 | } else unreachable; | 6137 | // } else unreachable; |
| 6877 | }; | 6138 | // }; |
| 6878 | break :base_address self.getSection(.{ | 6139 | // break :base_address self.getSection(.{ |
| 6879 | .seg = self.data_segment_cmd_index.?, | 6140 | // .seg = self.data_segment_cmd_index.?, |
| 6880 | .sect = sect_id, | 6141 | // .sect = sect_id, |
| 6881 | }).addr; | 6142 | // }).addr; |
| 6882 | } else 0; | 6143 | // } else 0; |
| 6883 | break :blk target_sym.n_value - base_address; | 6144 | // break :blk target_sym.n_value - base_address; |
| 6884 | }; | 6145 | // }; |
| 6885 | | 6146 | |
| 6886 | relocs.appendAssumeCapacity(.{ | 6147 | // relocs.appendAssumeCapacity(.{ |
| 6887 | .address = source_addr, | 6148 | // .address = source_addr, |
| 6888 | .tag = .relocation, | 6149 | // .tag = .relocation, |
| 6889 | .payload = .{ .target = target_addr }, | 6150 | // .payload = .{ .target = target_addr }, |
| 6890 | }); | 6151 | // }); |
| 6891 | } | 6152 | // } |
| 6892 | | 6153 | |
| 6893 | if (atom.contained.items.len == 0) { | 6154 | // if (atom.contained.items.len == 0) { |
| 6894 | try nodes.appendSlice(relocs.items); | 6155 | // try nodes.appendSlice(relocs.items); |
| 6895 | } else { | 6156 | // } else { |
| 6896 | // Need to reverse iteration order of relocs since by default for relocatable sources | 6157 | // // Need to reverse iteration order of relocs since by default for relocatable sources |
| 6897 | // they come in reverse. For linking, this doesn't matter in any way, however, for | 6158 | // // they come in reverse. For linking, this doesn't matter in any way, however, for |
| 6898 | // arranging the memoryline for displaying it does. | 6159 | // // arranging the memoryline for displaying it does. |
| 6899 | std.mem.reverse(Snapshot.Node, relocs.items); | 6160 | // std.mem.reverse(Snapshot.Node, relocs.items); |
| 6900 | | 6161 | |
| 6901 | var next_i: usize = 0; | 6162 | // var next_i: usize = 0; |
| 6902 | var last_rel: usize = 0; | 6163 | // var last_rel: usize = 0; |
| 6903 | while (next_i < atom.contained.items.len) : (next_i += 1) { | 6164 | // while (next_i < atom.contained.items.len) : (next_i += 1) { |
| 6904 | const loc = SymbolWithLoc{ | 6165 | // const loc = SymbolWithLoc{ |
| 6905 | .sym_index = atom.contained.items[next_i].sym_index, | 6166 | // .sym_index = atom.contained.items[next_i].sym_index, |
| 6906 | .file = atom.file, | 6167 | // .file = atom.file, |
| 6907 | }; | 6168 | // }; |
| 6908 | const cont_sym = self.getSymbol(loc); | 6169 | // const cont_sym = self.getSymbol(loc); |
| 6909 | const cont_sym_name = self.getSymbolName(loc); | 6170 | // const cont_sym_name = self.getSymbolName(loc); |
| 6910 | var contained_node = Snapshot.Node{ | 6171 | // var contained_node = Snapshot.Node{ |
| 6911 | .address = cont_sym.n_value, | 6172 | // .address = cont_sym.n_value, |
| 6912 | .tag = .atom_start, | 6173 | // .tag = .atom_start, |
| 6913 | .payload = .{ | 6174 | // .payload = .{ |
| 6914 | .name = cont_sym_name, | 6175 | // .name = cont_sym_name, |
| 6915 | .is_global = self.globals.contains(cont_sym_name), | 6176 | // .is_global = self.globals.contains(cont_sym_name), |
| 6916 | }, | 6177 | // }, |
| 6917 | }; | 6178 | // }; |
| 6918 | | 6179 | |
| 6919 | // Accumulate aliases | 6180 | // // Accumulate aliases |
| 6920 | var inner_aliases = std.ArrayList([]const u8).init(arena); | 6181 | // var inner_aliases = std.ArrayList([]const u8).init(arena); |
| 6921 | while (true) { | 6182 | // while (true) { |
| 6922 | if (next_i + 1 >= atom.contained.items.len) break; | 6183 | // if (next_i + 1 >= atom.contained.items.len) break; |
| 6923 | const next_sym_loc = SymbolWithLoc{ | 6184 | // const next_sym_loc = SymbolWithLoc{ |
| 6924 | .sym_index = atom.contained.items[next_i + 1].sym_index, | 6185 | // .sym_index = atom.contained.items[next_i + 1].sym_index, |
| 6925 | .file = atom.file, | 6186 | // .file = atom.file, |
| 6926 | }; | 6187 | // }; |
| 6927 | const next_sym = self.getSymbol(next_sym_loc); | 6188 | // const next_sym = self.getSymbol(next_sym_loc); |
| 6928 | if (next_sym.n_value != cont_sym.n_value) break; | 6189 | // if (next_sym.n_value != cont_sym.n_value) break; |
| 6929 | const next_sym_name = self.getSymbolName(next_sym_loc); | 6190 | // const next_sym_name = self.getSymbolName(next_sym_loc); |
| 6930 | if (self.globals.contains(next_sym_name)) { | 6191 | // if (self.globals.contains(next_sym_name)) { |
| 6931 | try inner_aliases.append(contained_node.payload.name); | 6192 | // try inner_aliases.append(contained_node.payload.name); |
| 6932 | contained_node.payload.name = next_sym_name; | 6193 | // contained_node.payload.name = next_sym_name; |
| 6933 | contained_node.payload.is_global = true; | 6194 | // contained_node.payload.is_global = true; |
| 6934 | } else try inner_aliases.append(next_sym_name); | 6195 | // } else try inner_aliases.append(next_sym_name); |
| 6935 | next_i += 1; | 6196 | // next_i += 1; |
| 6936 | } | 6197 | // } |
| 6937 | | 6198 | |
| 6938 | const cont_size = if (next_i + 1 < atom.contained.items.len) | 6199 | // const cont_size = if (next_i + 1 < atom.contained.items.len) |
| 6939 | self.getSymbol(.{ | 6200 | // self.getSymbol(.{ |
| 6940 | .sym_index = atom.contained.items[next_i + 1].sym_index, | 6201 | // .sym_index = atom.contained.items[next_i + 1].sym_index, |
| 6941 | .file = atom.file, | 6202 | // .file = atom.file, |
| 6942 | }).n_value - cont_sym.n_value | 6203 | // }).n_value - cont_sym.n_value |
| 6943 | else | 6204 | // else |
| 6944 | atom_sym.n_value + atom.size - cont_sym.n_value; | 6205 | // atom_sym.n_value + atom.size - cont_sym.n_value; |
| 6945 | | 6206 | |
| 6946 | contained_node.payload.aliases = inner_aliases.toOwnedSlice(); | 6207 | // contained_node.payload.aliases = inner_aliases.toOwnedSlice(); |
| 6947 | try nodes.append(contained_node); | 6208 | // try nodes.append(contained_node); |
| 6948 | | 6209 | |
| 6949 | for (relocs.items[last_rel..]) |rel| { | 6210 | // for (relocs.items[last_rel..]) |rel| { |
| 6950 | if (rel.address >= cont_sym.n_value + cont_size) { | 6211 | // if (rel.address >= cont_sym.n_value + cont_size) { |
| 6951 | break; | 6212 | // break; |
| 6952 | } | 6213 | // } |
| 6953 | try nodes.append(rel); | 6214 | // try nodes.append(rel); |
| 6954 | last_rel += 1; | 6215 | // last_rel += 1; |
| 6955 | } | 6216 | // } |
| 6956 | | 6217 | |
| 6957 | try nodes.append(.{ | 6218 | // try nodes.append(.{ |
| 6958 | .address = cont_sym.n_value + cont_size, | 6219 | // .address = cont_sym.n_value + cont_size, |
| 6959 | .tag = .atom_end, | 6220 | // .tag = .atom_end, |
| 6960 | .payload = .{}, | 6221 | // .payload = .{}, |
| 6961 | }); | 6222 | // }); |
| 6962 | } | 6223 | // } |
| 6963 | } | 6224 | // } |
| 6964 | | 6225 | |
| 6965 | try nodes.append(.{ | 6226 | // try nodes.append(.{ |
| 6966 | .address = atom_sym.n_value + atom.size, | 6227 | // .address = atom_sym.n_value + atom.size, |
| 6967 | .tag = .atom_end, | 6228 | // .tag = .atom_end, |
| 6968 | .payload = .{}, | 6229 | // .payload = .{}, |
| 6969 | }); | 6230 | // }); |
| 6970 | | 6231 | |
| 6971 | if (atom.next) |next| { | 6232 | // if (atom.next) |next| { |
| 6972 | atom = next; | 6233 | // atom = next; |
| 6973 | } else break; | 6234 | // } else break; |
| 6974 | } | 6235 | // } |
| 6975 | | 6236 | |
| 6976 | try nodes.append(.{ | 6237 | // try nodes.append(.{ |
| 6977 | .address = sect.addr + sect.size, | 6238 | // .address = sect.addr + sect.size, |
| 6978 | .tag = .section_end, | 6239 | // .tag = .section_end, |
| 6979 | .payload = .{}, | 6240 | // .payload = .{}, |
| | 6241 | // }); |
| | 6242 | // } |
| | 6243 | |
| | 6244 | // snapshot.nodes = nodes.toOwnedSlice(); |
| | 6245 | |
| | 6246 | // try std.json.stringify(snapshot, .{}, writer); |
| | 6247 | // try writer.writeByte(']'); |
| | 6248 | // } |
| | 6249 | |
| | 6250 | fn logSections(self: *MachO) void { |
| | 6251 | log.debug("sections:", .{}); |
| | 6252 | for (self.sections.items(.header)) |header, i| { |
| | 6253 | log.debug(" sect({d}): {s},{s} @{x}, sizeof({x})", .{ |
| | 6254 | i + 1, |
| | 6255 | header.segName(), |
| | 6256 | header.sectName(), |
| | 6257 | header.offset, |
| | 6258 | header.size, |
| 6980 | }); | 6259 | }); |
| 6981 | } | 6260 | } |
| 6982 | | | |
| 6983 | snapshot.nodes = nodes.toOwnedSlice(); | | |
| 6984 | | | |
| 6985 | try std.json.stringify(snapshot, .{}, writer); | | |
| 6986 | try writer.writeByte(']'); | | |
| 6987 | } | 6261 | } |
| 6988 | | 6262 | |
| 6989 | fn logSymAttributes(sym: macho.nlist_64, buf: *[9]u8) []const u8 { | 6263 | fn logSymAttributes(sym: macho.nlist_64, buf: *[9]u8) []const u8 { |
| ... | @@ -7104,26 +6378,19 @@ fn logSymtab(self: *MachO) void { | ... | @@ -7104,26 +6378,19 @@ fn logSymtab(self: *MachO) void { |
| 7104 | } | 6378 | } |
| 7105 | } | 6379 | } |
| 7106 | | 6380 | |
| 7107 | fn logSectionOrdinals(self: *MachO) void { | | |
| 7108 | for (self.section_ordinals.keys()) |match, i| { | | |
| 7109 | const sect = self.getSection(match); | | |
| 7110 | log.debug("sect({d}, '{s},{s}')", .{ i + 1, sect.segName(), sect.sectName() }); | | |
| 7111 | } | | |
| 7112 | } | | |
| 7113 | | | |
| 7114 | fn logAtoms(self: *MachO) void { | 6381 | fn logAtoms(self: *MachO) void { |
| 7115 | log.debug("atoms:", .{}); | 6382 | log.debug("atoms:", .{}); |
| 7116 | var it = self.atoms.iterator(); | 6383 | |
| 7117 | while (it.next()) |entry| { | 6384 | const slice = self.sections.slice(); |
| 7118 | const match = entry.key_ptr.*; | 6385 | for (slice.items(.last_atom)) |last, i| { |
| 7119 | var atom = entry.value_ptr.*; | 6386 | var atom = last orelse continue; |
| | 6387 | const header = slice.items(.header)[i]; |
| 7120 | | 6388 | |
| 7121 | while (atom.prev) |prev| { | 6389 | while (atom.prev) |prev| { |
| 7122 | atom = prev; | 6390 | atom = prev; |
| 7123 | } | 6391 | } |
| 7124 | | 6392 | |
| 7125 | const sect = self.getSection(match); | 6393 | log.debug("{s},{s}", .{ header.segName(), header.sectName() }); |
| 7126 | log.debug("{s},{s}", .{ sect.segName(), sect.sectName() }); | | |
| 7127 | | 6394 | |
| 7128 | while (true) { | 6395 | while (true) { |
| 7129 | self.logAtom(atom); | 6396 | self.logAtom(atom); |